[Bugfix] fix issue 5065 (#5118)
This commit is contained in:
1 parent
a432b8a325
commit
238fc0a609
1 file changed
+1
-1
@@ -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]
|
||||
|
||||
Reference in new issue
Block a user