Spaces:

mikeee
/

chatglm2-6b-test

Runtime error

mikeee commited on Jul 14, 2023

Commit

dcea2af

1 Parent(s): 89cb869

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,10 +1,11 @@
 import os
-import time
 # os.system("pip install --upgrade torch transformers sentencepiece scipy cpm_kernels accelerate bitsandbytes loguru")
 os.system("pip install torch transformers sentencepiece loguru")
 import gradio as gr
 from transformers import AutoTokenizer, AutoModel, AutoModelForCausalLM
 # fix timezone in Linux

 import os
 # os.system("pip install --upgrade torch transformers sentencepiece scipy cpm_kernels accelerate bitsandbytes loguru")
 os.system("pip install torch transformers sentencepiece loguru")
+logger.debug("load")
 import gradio as gr
+import torch
 from transformers import AutoTokenizer, AutoModel, AutoModelForCausalLM
 # fix timezone in Linux