Update app.py

app.py

@@ -267,8 +267,8 @@ class LTX23DistilledA2VPipeline(DistilledPipeline):
 # Model repos
 LTX_MODEL_REPO = "Lightricks/LTX-2.3"
 GEMMA_REPO ="Lightricks/gemma-3-12b-it-qat-q4_0-unquantized"
-GEMMA_ABLITERATED_REPO = "FusionCow/Gemma-3-12b-Abliterated-LTX2"
-GEMMA_ABLITERATED_FILE = "gemma_ablit_fixed_bf16.safetensors"
+GEMMA_ABLITERATED_REPO = "Sikaworld1990/gemma-3-12b-it-abliterated-sikaworld-high-fidelity-edition-Ltx-2"
+GEMMA_ABLITERATED_FILE = "gemma-3-12b-it-abliterated-sikaworld-high-fidelity-edition.safetensors"
 
 # Download model checkpoints
 print("=" * 80)
@@ -338,10 +338,12 @@ else:
             filename=shard_name
         )
 
+    _fp8_types = {torch.float8_e4m3fn, torch.float8_e5m2}
     raw = load_file(abliterated_weights_path)
     merged = {}
     for key, tensor in raw.items():
-        merged[f"language_model.{key}"] = tensor
+        t = tensor.to(torch.bfloat16) if tensor.dtype in _fp8_types else tensor
+        merged[f"language_model.{key}"] = t
     del raw
 
     for key, shard_name in vision_keys.items():