From 059ebeead7bd6e0c8193715998e21253d89261e3 Mon Sep 17 00:00:00 2001 From: Will Miao Date: Tue, 18 Nov 2025 14:16:22 +0800 Subject: [PATCH] feat: include Negative file type in primary file selection for embeddings --- py/services/download_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/services/download_manager.py b/py/services/download_manager.py index 17bf0e4d..1ece024d 100644 --- a/py/services/download_manager.py +++ b/py/services/download_manager.py @@ -331,7 +331,7 @@ class DownloadManager: await progress_callback(0) # 2. Get file information - file_info = next((f for f in version_info.get('files', []) if f.get('primary') and f.get('type') == 'Model'), None) + file_info = next((f for f in version_info.get('files', []) if f.get('primary') and f.get('type') in ('Model', 'Negative')), None) if not file_info: return {'success': False, 'error': 'No primary file found in metadata'} mirrors = file_info.get('mirrors') or []