Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ model_dict = {
|
|
| 102 |
}
|
| 103 |
|
| 104 |
models = {
|
| 105 |
-
k: StableMultiDiffusionPipeline(device,
|
| 106 |
for k, v in model_dict.items()
|
| 107 |
}
|
| 108 |
|
|
|
|
| 102 |
}
|
| 103 |
|
| 104 |
models = {
|
| 105 |
+
k: StableMultiDiffusionPipeline(device, sd_version='1.5', hf_key=v).cuda()
|
| 106 |
for k, v in model_dict.items()
|
| 107 |
}
|
| 108 |
|