dagloop5 commited on
Commit
2e195e9
·
verified ·
1 Parent(s): 7ae5675

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- stepper = Res2sDiffusionStep()
175
- stage_2_stepper = EulerDiffusionStep()
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=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=stage_2_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,