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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -282,7 +282,7 @@ class LTX23DistilledA2VPipeline:
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,
@@ -316,7 +316,7 @@ class LTX23DistilledA2VPipeline:
316
  conditionings=stage_2_conditionings,
317
  noiser=noiser,
318
  sigmas=stage_2_sigmas,
319
- stepper=stepper,
320
  denoising_loop_fn=stage2_denoising_loop,
321
  components=self.pipeline_components,
322
  dtype=dtype,
 
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,
 
316
  conditionings=stage_2_conditionings,
317
  noiser=noiser,
318
  sigmas=stage_2_sigmas,
319
+ stepper=euler_stepper,
320
  denoising_loop_fn=stage2_denoising_loop,
321
  components=self.pipeline_components,
322
  dtype=dtype,