Skip to content

Commit b87e482

Browse files
committed
新增:跳过扫描多余文件夹
1 parent bdec072 commit b87e482

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

model_manager.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,15 @@ def _get_file_mtime(self, file_path: Path) -> float:
8484
async def scan_models(self):
8585
"""扫描指定目录下的所有.safetensors文件"""
8686
print(f"开始扫描目录: {self.models_path}")
87-
safetensors_files = list(self.models_path.rglob("*.safetensors"))
87+
88+
# 只扫描 checkpoints 和 loras 文件夹
89+
allowed_folders = ['checkpoints', 'loras']
90+
safetensors_files = []
91+
92+
for folder in allowed_folders:
93+
folder_path = self.models_path / folder
94+
if folder_path.exists():
95+
safetensors_files.extend(list(folder_path.rglob("*.safetensors")))
8896

8997
if not safetensors_files:
9098
print("未找到任何.safetensors文件")

0 commit comments

Comments
 (0)