Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -171,7 +171,7 @@ class LTX23DistilledA2VPipeline:
|
|
| 171 |
|
| 172 |
generator = torch.Generator(device=self.device).manual_seed(seed)
|
| 173 |
noiser = GaussianNoiser(generator=generator)
|
| 174 |
-
|
| 175 |
euler_stepper = EulerDiffusionStep()
|
| 176 |
dtype = torch.bfloat16
|
| 177 |
|
|
@@ -229,7 +229,7 @@ class LTX23DistilledA2VPipeline:
|
|
| 229 |
sigmas=sigmas,
|
| 230 |
video_state=video_state,
|
| 231 |
audio_state=audio_state,
|
| 232 |
-
stepper=
|
| 233 |
denoise_fn=multi_modal_guider_denoising_func(
|
| 234 |
video_guider=MultiModalGuider(
|
| 235 |
params=video_guider_params,
|
|
@@ -282,7 +282,7 @@ class LTX23DistilledA2VPipeline:
|
|
| 282 |
conditionings=stage_1_conditionings,
|
| 283 |
noiser=noiser,
|
| 284 |
sigmas=stage_1_sigmas,
|
| 285 |
-
stepper=
|
| 286 |
denoising_loop_fn=stage1_denoising_loop,
|
| 287 |
components=self.pipeline_components,
|
| 288 |
dtype=dtype,
|
|
|
|
| 171 |
|
| 172 |
generator = torch.Generator(device=self.device).manual_seed(seed)
|
| 173 |
noiser = GaussianNoiser(generator=generator)
|
| 174 |
+
stepper = Res2sDiffusionStep()
|
| 175 |
euler_stepper = EulerDiffusionStep()
|
| 176 |
dtype = torch.bfloat16
|
| 177 |
|
|
|
|
| 229 |
sigmas=sigmas,
|
| 230 |
video_state=video_state,
|
| 231 |
audio_state=audio_state,
|
| 232 |
+
stepper=stepper,
|
| 233 |
denoise_fn=multi_modal_guider_denoising_func(
|
| 234 |
video_guider=MultiModalGuider(
|
| 235 |
params=video_guider_params,
|
|
|
|
| 282 |
conditionings=stage_1_conditionings,
|
| 283 |
noiser=noiser,
|
| 284 |
sigmas=stage_1_sigmas,
|
| 285 |
+
stepper=stepper,
|
| 286 |
denoising_loop_fn=stage1_denoising_loop,
|
| 287 |
components=self.pipeline_components,
|
| 288 |
dtype=dtype,
|