Spaces:
Running
Running
Commit
·
cc69909
1
Parent(s):
2e62db4
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,8 +26,8 @@ class Model:
|
|
| 26 |
self.code = code
|
| 27 |
|
| 28 |
model = [
|
| 29 |
-
Model("DDIMScheduler", "
|
| 30 |
-
Model("PNDMScheduler", "
|
| 31 |
|
| 32 |
current_model = model[0]
|
| 33 |
|
|
|
|
| 26 |
self.code = code
|
| 27 |
|
| 28 |
model = [
|
| 29 |
+
Model("DDIMScheduler", "DDIMScheduler.from_pretrained(pipeline_name)"),
|
| 30 |
+
Model("PNDMScheduler", "PNDMScheduler.from_pretrained(pipeline_name)")]
|
| 31 |
|
| 32 |
current_model = model[0]
|
| 33 |
|