NicolasG2523 commited on
Commit
12f7db7
·
verified ·
1 Parent(s): 74ce720

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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