update import

main
JustinLin610 1 year ago
parent 071a4af365
commit 50131acda4

@ -9,7 +9,8 @@ from argparse import ArgumentParser
import gradio as gr import gradio as gr
import mdtex2html import mdtex2html
import torch.cuda
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig from transformers.generation import GenerationConfig

Loading…
Cancel
Save