Update app.py
Browse files
app.py
CHANGED
|
@@ -171,8 +171,8 @@ class LTX23DistilledA2VPipeline:
|
|
| 171 |
|
| 172 |
generator = torch.Generator(device=self.device).manual_seed(seed)
|
| 173 |
noiser = GaussianNoiser(generator=generator)
|
| 174 |
-
|
| 175 |
-
|
| 176 |
dtype = torch.bfloat16
|
| 177 |
|
| 178 |
ctx_p, ctx_n = encode_prompts(
|
|
@@ -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,
|
|
@@ -250,7 +250,7 @@ class LTX23DistilledA2VPipeline:
|
|
| 250 |
sigmas=sigmas,
|
| 251 |
video_state=video_state,
|
| 252 |
audio_state=audio_state,
|
| 253 |
-
stepper=
|
| 254 |
denoise_fn=simple_denoising_func(
|
| 255 |
video_context=v_context_p,
|
| 256 |
audio_context=a_context_p,
|
|
|
|
| 171 |
|
| 172 |
generator = torch.Generator(device=self.device).manual_seed(seed)
|
| 173 |
noiser = GaussianNoiser(generator=generator)
|
| 174 |
+
res2s_stepper = Res2sDiffusionStep()
|
| 175 |
+
euler_stepper = EulerDiffusionStep()
|
| 176 |
dtype = torch.bfloat16
|
| 177 |
|
| 178 |
ctx_p, ctx_n = encode_prompts(
|
|
|
|
| 229 |
sigmas=sigmas,
|
| 230 |
video_state=video_state,
|
| 231 |
audio_state=audio_state,
|
| 232 |
+
stepper=res2s_stepper,
|
| 233 |
denoise_fn=multi_modal_guider_denoising_func(
|
| 234 |
video_guider=MultiModalGuider(
|
| 235 |
params=video_guider_params,
|
|
|
|
| 250 |
sigmas=sigmas,
|
| 251 |
video_state=video_state,
|
| 252 |
audio_state=audio_state,
|
| 253 |
+
stepper=euler_stepper,
|
| 254 |
denoise_fn=simple_denoising_func(
|
| 255 |
video_context=v_context_p,
|
| 256 |
audio_context=a_context_p,
|