移除torch依赖

app.py

@@ -10,7 +10,7 @@ from argparse import ArgumentParser
 import gradio as gr
 import mdtex2html
 
-import torch
+# import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation import GenerationConfig
 
@@ -110,8 +110,8 @@ def _parse_text(text):
 def _gc():
     import gc
     gc.collect()
-    if torch.cuda.is_available():
-        torch.cuda.empty_cache()
+    # if torch.cuda.is_available():
+    #     torch.cuda.empty_cache()
 
 
 def _launch_demo(args, model, tokenizer, config):