Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,7 +88,7 @@ print("Diffusion path: ", diffusion_path)
|
|
| 88 |
# Initialize model + pipeline
|
| 89 |
disable_torch_init()
|
| 90 |
tokenizer, multi_model, _ = load_pretrained_model(
|
| 91 |
-
|
| 92 |
)
|
| 93 |
pipe = DiffusionPipeline.from_pretrained(
|
| 94 |
diffusion_path,
|
|
|
|
| 88 |
# Initialize model + pipeline
|
| 89 |
disable_torch_init()
|
| 90 |
tokenizer, multi_model, _ = load_pretrained_model(
|
| 91 |
+
diffusion_path, None, get_model_name_from_path(diffusion_path)
|
| 92 |
)
|
| 93 |
pipe = DiffusionPipeline.from_pretrained(
|
| 94 |
diffusion_path,
|