pull/637/head
zhengjie.xu 2024-01-19 17:16:08 +08:00 committed by GitHub
parent 19f8eb4534
commit 86c5da6dd5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -17,7 +17,7 @@ from typing import Callable, List, Optional
import streamlit as st import streamlit as st
import torch import torch
from torch import nn from torch import nn
from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip
from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList
from transformers.utils import logging from transformers.utils import logging