update import

1 year ago · 50131acda4
parent 071a4af365
commit 50131acda4
1 changed files with 2 additions and 1 deletions
--- a/web_demo.py
+++ b/web_demo.py
@ -9,7 +9,8 @@ from argparse import ArgumentParser

 import gradio as gr
 import mdtex2html
-import torch.cuda
+
+import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation import GenerationConfig