Spaces:
Runtime error
Runtime error
Commit
·
fa69421
1
Parent(s):
13b14ac
fix bugs
Browse files- animatediff/pipelines/pipeline_animation.py +2 -2
- app.py +1 -6
animatediff/pipelines/pipeline_animation.py
CHANGED
|
@@ -474,7 +474,7 @@ class AnimationFreeInitPipeline(AnimationPipeline):
|
|
| 474 |
filter_shape,
|
| 475 |
device=self._execution_device,
|
| 476 |
filter_type=filter_params.method,
|
| 477 |
-
n=filter_params.n,
|
| 478 |
d_s=filter_params.d_s,
|
| 479 |
d_t=filter_params.d_t
|
| 480 |
)
|
|
@@ -653,4 +653,4 @@ class AnimationFreeInitPipeline(AnimationPipeline):
|
|
| 653 |
if return_orig:
|
| 654 |
return AnimationFreeInitPipelineOutput(videos=video, orig_videos=orig_video)
|
| 655 |
|
| 656 |
-
return
|
|
|
|
| 474 |
filter_shape,
|
| 475 |
device=self._execution_device,
|
| 476 |
filter_type=filter_params.method,
|
| 477 |
+
n=filter_params.n if filter_params.method=="butterworth" else None,
|
| 478 |
d_s=filter_params.d_s,
|
| 479 |
d_t=filter_params.d_t
|
| 480 |
)
|
|
|
|
| 653 |
if return_orig:
|
| 654 |
return AnimationFreeInitPipelineOutput(videos=video, orig_videos=orig_video)
|
| 655 |
|
| 656 |
+
return AnimationPipelineOutput(videos=video)
|
app.py
CHANGED
|
@@ -216,7 +216,7 @@ class AnimateController:
|
|
| 216 |
|
| 217 |
vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
|
| 218 |
|
| 219 |
-
shape = [1, 4, 16, self.
|
| 220 |
self.freq_filter = get_freq_filter(
|
| 221 |
shape,
|
| 222 |
device="cuda",
|
|
@@ -254,11 +254,6 @@ class AnimateController:
|
|
| 254 |
if self.selected_base_model != base_model_dropdown: self.update_base_model(base_model_dropdown)
|
| 255 |
if self.selected_motion_module != motion_module_dropdown: self.update_motion_module(motion_module_dropdown)
|
| 256 |
|
| 257 |
-
self.set_width = width_slider
|
| 258 |
-
self.set_height = height_slider
|
| 259 |
-
self.selected_filter_type = filter_type_dropdown
|
| 260 |
-
self.set_d_s = d_s
|
| 261 |
-
self.set_d_t = d_t
|
| 262 |
if self.set_width != width_slider or self.set_height != height_slider or self.selected_filter_type != filter_type_dropdown or self.set_d_s != d_s or self.set_d_t != d_t:
|
| 263 |
self.update_filter(width_slider, height_slider, filter_type_dropdown, d_s, d_t)
|
| 264 |
|
|
|
|
| 216 |
|
| 217 |
vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
|
| 218 |
|
| 219 |
+
shape = [1, 4, 16, self.set_height//vae_scale_factor, self.set_width//vae_scale_factor]
|
| 220 |
self.freq_filter = get_freq_filter(
|
| 221 |
shape,
|
| 222 |
device="cuda",
|
|
|
|
| 254 |
if self.selected_base_model != base_model_dropdown: self.update_base_model(base_model_dropdown)
|
| 255 |
if self.selected_motion_module != motion_module_dropdown: self.update_motion_module(motion_module_dropdown)
|
| 256 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 257 |
if self.set_width != width_slider or self.set_height != height_slider or self.selected_filter_type != filter_type_dropdown or self.set_d_s != d_s or self.set_d_t != d_t:
|
| 258 |
self.update_filter(width_slider, height_slider, filter_type_dropdown, d_s, d_t)
|
| 259 |
|