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
|