[Bugfix] fix issue 5065 (#5118)

This commit is contained in:
zhumingjue138 authored and GitHub committed 2026-07-15 19:39:44 -07:00
1 parent a432b8a325
commit 238fc0a609
1 file changed
+1 -1
+1 -1
View File
@@ -22,7 +22,7 @@ def tiny_model_paths(request, run_level):
if not hasattr(item, "callspec"):
continue
model_name = item.callspec.params.get("model_name")
if model_name is None:
if model_name is None or model_name not in DIFFUSION_TEST_SETTINGS:
continue
if model_name not in model_paths:
settings = DIFFUSION_TEST_SETTINGS[model_name]