dagloop5 commited on
Commit
8ce4a02
·
verified ·
1 Parent(s): d673b48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -274,7 +274,7 @@ weights_dir = Path("weights")
274
  weights_dir.mkdir(exist_ok=True)
275
  checkpoint_path = hf_hub_download(
276
  repo_id="SulphurAI/Sulphur-2-base",
277
- filename="sulphur_dev_fp8mixed.safetensors",
278
  local_dir=str(weights_dir),
279
  local_dir_use_symlinks=False,
280
  )
@@ -334,7 +334,7 @@ pipeline = LTX23DistilledA2VPipeline(
334
  spatial_upsampler_path=spatial_upsampler_path,
335
  gemma_root=gemma_root,
336
  loras=[],
337
- quantization=QuantizationPolicy.fp8_cast(), # keep FP8 quantization unchanged
338
  )
339
  # ----------------------------------------------------------------
340
 
 
274
  weights_dir.mkdir(exist_ok=True)
275
  checkpoint_path = hf_hub_download(
276
  repo_id="SulphurAI/Sulphur-2-base",
277
+ filename="sulphur_distil_bf16.safetensors",
278
  local_dir=str(weights_dir),
279
  local_dir_use_symlinks=False,
280
  )
 
334
  spatial_upsampler_path=spatial_upsampler_path,
335
  gemma_root=gemma_root,
336
  loras=[],
337
+ quantization=None, # keep FP8 quantization unchanged
338
  )
339
  # ----------------------------------------------------------------
340