Update app.py
Browse files
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", "
|
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
|