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