oh-my-dear-ai commited on
Commit
691b49c
·
verified ·
1 Parent(s): 64aeca7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ is_share = os.environ.get("is_share", "False")
48
  is_share = eval(is_share)
49
  if "_CUDA_VISIBLE_DEVICES" in os.environ:
50
  os.environ["CUDA_VISIBLE_DEVICES"] = os.environ["_CUDA_VISIBLE_DEVICES"]
51
- is_half = eval(os.environ.get("is_half", "True"))
52
  import gradio as gr
53
  from transformers import AutoModelForMaskedLM, AutoTokenizer
54
  import numpy as np
 
48
  is_share = eval(is_share)
49
  if "_CUDA_VISIBLE_DEVICES" in os.environ:
50
  os.environ["CUDA_VISIBLE_DEVICES"] = os.environ["_CUDA_VISIBLE_DEVICES"]
51
+ is_half = eval(os.environ.get("is_half", "False")) # 部署到hg前要检查是否cpu部署(is_half改为false)
52
  import gradio as gr
53
  from transformers import AutoModelForMaskedLM, AutoTokenizer
54
  import numpy as np