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 torch
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.utils import logging