Spaces:
Runtime error
Runtime error
fix debug
Browse files- inference/pipeline.py +2 -1
inference/pipeline.py
CHANGED
@@ -59,7 +59,8 @@ class RealCustomInferencePipeline:
|
|
59 |
repo_type="model",
|
60 |
local_dir="./", # 指定本地目录
|
61 |
allow_patterns="ckpts/**", # 只下载 ckpts 文件夹内容
|
62 |
-
local_dir_use_symlinks=False # 直接存储文件而非符号链接
|
|
|
63 |
)
|
64 |
|
65 |
self.device = device
|
|
|
59 |
repo_type="model",
|
60 |
local_dir="./", # 指定本地目录
|
61 |
allow_patterns="ckpts/**", # 只下载 ckpts 文件夹内容
|
62 |
+
local_dir_use_symlinks=False, # 直接存储文件而非符号链接
|
63 |
+
request_timeout=600 # 设置超时时间为600秒(10分钟)
|
64 |
)
|
65 |
|
66 |
self.device = device
|