Spaces:
Running
on
T4
Running
on
T4
Update gradio_app.py
Browse files- gradio_app.py +16 -0
gradio_app.py
CHANGED
|
@@ -12,6 +12,22 @@
|
|
| 12 |
# fine-tuning enabling code and other elements of the foregoing made publicly available
|
| 13 |
# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
import os
|
| 16 |
import random
|
| 17 |
import shutil
|
|
|
|
| 12 |
# fine-tuning enabling code and other elements of the foregoing made publicly available
|
| 13 |
# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
|
| 14 |
|
| 15 |
+
# --- Fix for OMP_NUM_THREADS '3500m' and Hugging Face Space ---
|
| 16 |
+
import os, re
|
| 17 |
+
val = os.environ.get("OMP_NUM_THREADS", "")
|
| 18 |
+
m = re.match(r"^(\d+)$", str(val))
|
| 19 |
+
if not m:
|
| 20 |
+
# default seguro
|
| 21 |
+
try:
|
| 22 |
+
import multiprocessing as mp
|
| 23 |
+
os.environ["OMP_NUM_THREADS"] = str(max(1, min(8, mp.cpu_count())))
|
| 24 |
+
except Exception:
|
| 25 |
+
os.environ["OMP_NUM_THREADS"] = "4"
|
| 26 |
+
os.environ.setdefault("OMP_DYNAMIC", "TRUE")
|
| 27 |
+
os.environ.setdefault("HF_HUB_ENABLE_HF_TRANSFER", "1")
|
| 28 |
+
os.environ.setdefault("HF_HOME", "/home/user/.cache/huggingface")
|
| 29 |
+
os.environ.setdefault("PYMESHLAB_NO_PLUGINS", "1")
|
| 30 |
+
# ---------------------------------------------------------------
|
| 31 |
import os
|
| 32 |
import random
|
| 33 |
import shutil
|