From 50131acda4cbb2a03dea36be6718265af4672507 Mon Sep 17 00:00:00 2001
From: JustinLin610 <justinlin930319@hotmail.com>
Date: Sun, 27 Aug 2023 18:30:48 +0800
Subject: [PATCH] update import

---
 web_demo.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/web_demo.py b/web_demo.py
index 71d3257..922df00 100755
--- 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