Spaces:
Running
on
Zero
Running
on
Zero
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,
|