only use .components if available, so we can use earlier diffusers versions

gadicc · gadicc · commit db361f0536bf · 2022-10-24T10:00:27.000+01:00
diff --git a/app.py b/app.py
@@ -35,7 +35,18 @@
 def createPipelinesFromModel(model):
     pipelines = dict()
     for pipeline in PIPELINES:
-        pipelines[pipeline] = getattr(_pipelines, pipeline)(**model.components)
+        if hasattr(model, "components"):
+            pipelines[pipeline] = getattr(_pipelines, pipeline)(**model.components)
+        else:
+            pipelines[pipeline] = getattr(_pipelines, pipeline)(
+                vae=model.vae,
+                text_encoder=model.text_encoder,
+                tokenizer=model.tokenizer,
+                unet=model.unet,
+                scheduler=model.scheduler,
+                safety_checker=model.safety_checker,
+                feature_extractor=model.feature_extractor,
+            )
     return pipelines
 
 
