fffiloni commited on
Commit
c3d106b
·
verified ·
1 Parent(s): 096e320

Update auffusion_pipeline.py

Browse files
Files changed (1) hide show
  1. 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
- vocoder=vocoder if vocoder is not None else None,
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