Spaces:
Running
on
Zero
Running
on
Zero
Update auffusion_pipeline.py
Browse files- auffusion_pipeline.py +2 -1
auffusion_pipeline.py
CHANGED
|
@@ -487,7 +487,7 @@ class AuffusionPipeline(DiffusionPipeline):
|
|
| 487 |
scheduler=scheduler,
|
| 488 |
safety_checker=safety_checker,
|
| 489 |
feature_extractor=feature_extractor,
|
| 490 |
-
|
| 491 |
)
|
| 492 |
|
| 493 |
# Register config-only (non-module) components — avoids ValueError during .to()
|
|
@@ -497,6 +497,7 @@ class AuffusionPipeline(DiffusionPipeline):
|
|
| 497 |
tokenizer_list=tokenizer_list,
|
| 498 |
adapter_list=adapter_list,
|
| 499 |
tokenizer_model_max_length=tokenizer_model_max_length,
|
|
|
|
| 500 |
)
|
| 501 |
|
| 502 |
# Other logic
|
|
|
|
| 487 |
scheduler=scheduler,
|
| 488 |
safety_checker=safety_checker,
|
| 489 |
feature_extractor=feature_extractor,
|
| 490 |
+
|
| 491 |
)
|
| 492 |
|
| 493 |
# Register config-only (non-module) components — avoids ValueError during .to()
|
|
|
|
| 497 |
tokenizer_list=tokenizer_list,
|
| 498 |
adapter_list=adapter_list,
|
| 499 |
tokenizer_model_max_length=tokenizer_model_max_length,
|
| 500 |
+
vocoder=vocoder if vocoder is not None else None,
|
| 501 |
)
|
| 502 |
|
| 503 |
# Other logic
|