dagloop5 commited on
Commit
5073ab4
·
verified ·
1 Parent(s): d824bed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- res2s_stepper = Res2sDiffusionStep()
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=res2s_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=res2s_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,