bluestarburst
/

AnimateDiff-SceneFusion

Model card Files Files and versions

bluestarburst commited on Nov 24, 2023

Commit

f8bc965

·

1 Parent(s): ddb7519

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -30,7 +30,7 @@ class EndpointHandler():
         self.pipeline = AnimationPipeline(
             vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet,
-            scheduler=DDIMScheduler(**OmegaConf.to_container(inference_config.noise_scheduler_kwargs)),
         ).to("cuda")
         self.pipeline = load_weights(

         self.pipeline = AnimationPipeline(
             vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet,
+            scheduler=DDIMScheduler(**OmegaConf.to_container(inference_config.noise_scheduler_kwargs.DDIMScheduler)),
         ).to("cuda")
         self.pipeline = load_weights(