Add new keys to weights skip

nod-ai · Jun 21, 2024 · 3786061 · 3786061
1 parent fd2185b
commit 3786061
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/turbine_models/custom_models/sdxl_inference/sdxl_compiled_pipeline.py b/models/turbine_models/custom_models/sdxl_inference/sdxl_compiled_pipeline.py
@@ -220,7 +220,7 @@ def is_prepared(self, vmfbs, weights):
                 missing.append(key + " vmfb")
 
         for w_key in weights:
-            if any(x in w_key for x in ["pipeline", "scheduler"]) or (
+            if any(x in w_key for x in ["fullpipeline", "unetloop", "scheduler"]) or (
                 self.external_weights is None
             ):
                 continue