GOGO198 commited on
Commit
252fd1e
·
verified ·
1 Parent(s): 015c037

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ os.makedirs(CACHE_DIR, exist_ok=True)
14
  # 减少内存占用
15
  # os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:32"
16
  # torch.set_num_threads(1)
17
- os.environ["OMP_NUM_THREADS"] = "8"
 
18
 
19
  # 全局变量 - 移除了clip_model和clip_processor
20
  index = None
 
14
  # 减少内存占用
15
  # os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:32"
16
  # torch.set_num_threads(1)
17
+ os.environ["OMP_NUM_THREADS"] = "2"
18
+ os.environ["TOKENIZERS_PARALLELISM"] = "false" # 防止tokenizer内存泄漏
19
 
20
  # 全局变量 - 移除了clip_model和clip_processor
21
  index = None