Update pipeline.py
Browse files- pipeline.py +2 -1
pipeline.py
CHANGED
|
@@ -844,6 +844,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 844 |
latent_mode: Optional[str] = "normal",
|
| 845 |
smooth_weight: Optional[float] = 0.5,
|
| 846 |
smooth_steps: Optional[int] = 3,
|
|
|
|
| 847 |
):
|
| 848 |
r"""
|
| 849 |
The call function to the pipeline for generation.
|
|
@@ -1066,7 +1067,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 1066 |
# Iterate over each index in the context group
|
| 1067 |
local_context_size = context_size
|
| 1068 |
if timestep <= 1:
|
| 1069 |
-
local_context_size = context_size *
|
| 1070 |
for index in range(local_context_size):
|
| 1071 |
# if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
|
| 1072 |
if timestep <= 1:
|
|
|
|
| 844 |
latent_mode: Optional[str] = "normal",
|
| 845 |
smooth_weight: Optional[float] = 0.5,
|
| 846 |
smooth_steps: Optional[int] = 3,
|
| 847 |
+
initial_context_scale: Optional[float] = 1.0,
|
| 848 |
):
|
| 849 |
r"""
|
| 850 |
The call function to the pipeline for generation.
|
|
|
|
| 1067 |
# Iterate over each index in the context group
|
| 1068 |
local_context_size = context_size
|
| 1069 |
if timestep <= 1:
|
| 1070 |
+
local_context_size = int(context_size * initial_context_scale)
|
| 1071 |
for index in range(local_context_size):
|
| 1072 |
# if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
|
| 1073 |
if timestep <= 1:
|