Spaces:
Sleeping
Sleeping
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 |
|