mikeee commited on
Commit
dcea2af
1 Parent(s): 89cb869

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,10 +1,11 @@
1
  import os
2
- import time
3
 
4
  # os.system("pip install --upgrade torch transformers sentencepiece scipy cpm_kernels accelerate bitsandbytes loguru")
5
  os.system("pip install torch transformers sentencepiece loguru")
6
 
 
7
  import gradio as gr
 
8
  from transformers import AutoTokenizer, AutoModel, AutoModelForCausalLM
9
 
10
  # fix timezone in Linux
 
1
  import os
 
2
 
3
  # os.system("pip install --upgrade torch transformers sentencepiece scipy cpm_kernels accelerate bitsandbytes loguru")
4
  os.system("pip install torch transformers sentencepiece loguru")
5
 
6
+ logger.debug("load")
7
  import gradio as gr
8
+ import torch
9
  from transformers import AutoTokenizer, AutoModel, AutoModelForCausalLM
10
 
11
  # fix timezone in Linux