dagloop5 commited on
Commit
4c881d2
·
verified ·
1 Parent(s): f4bdc10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -407,7 +407,7 @@ pipeline = LTX23DistilledA2VPipeline(
407
  spatial_upsampler_path=spatial_upsampler_path,
408
  gemma_root=gemma_root,
409
  loras=[],
410
- quantization=None,
411
  )
412
 
413
  def _make_lora_key(singularity_strength, teneros_strength, sulphur_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength, physics_strength, reasoning_strength, twostep_strength, mcfurry_strength, dm_strength, praxis_strength, threed_strength, concept_strength, bulge_strength) -> tuple[str, str]:
@@ -435,7 +435,7 @@ def prepare_lora_cache(
435
  return PENDING_LORA_STATUS
436
  try:
437
  progress(0.35, desc="Building fused CPU transformer")
438
- tmp_ledger = pipeline.model_ledger.__class__(dtype=ledger.dtype, device=torch.device("cpu"), checkpoint_path=str(checkpoint_path), spatial_upsampler_path=str(spatial_upsampler_path), gemma_root_path=str(gemma_root), loras=tuple(loras_for_builder), quantization=None)
439
  new_transformer_cpu = tmp_ledger.transformer()
440
  progress(0.70, desc="Extracting fused state_dict")
441
  state = {k: v.detach().cpu().contiguous() for k, v in new_transformer_cpu.state_dict().items()}
 
407
  spatial_upsampler_path=spatial_upsampler_path,
408
  gemma_root=gemma_root,
409
  loras=[],
410
+ quantization=QuantizationPolicy.fp8_cast(),
411
  )
412
 
413
  def _make_lora_key(singularity_strength, teneros_strength, sulphur_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength, physics_strength, reasoning_strength, twostep_strength, mcfurry_strength, dm_strength, praxis_strength, threed_strength, concept_strength, bulge_strength) -> tuple[str, str]:
 
435
  return PENDING_LORA_STATUS
436
  try:
437
  progress(0.35, desc="Building fused CPU transformer")
438
+ tmp_ledger = pipeline.model_ledger.__class__(dtype=ledger.dtype, device=torch.device("cpu"), checkpoint_path=str(checkpoint_path), spatial_upsampler_path=str(spatial_upsampler_path), gemma_root_path=str(gemma_root), loras=tuple(loras_for_builder), quantization=QuantizationPolicy.fp8_cast())
439
  new_transformer_cpu = tmp_ledger.transformer()
440
  progress(0.70, desc="Extracting fused state_dict")
441
  state = {k: v.detach().cpu().contiguous() for k, v in new_transformer_cpu.state_dict().items()}