CoreloneH commited on
Commit
0b318d9
·
1 Parent(s): fddee3b
Files changed (1) hide show
  1. 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