Update app.py
Browse files
app.py
CHANGED
|
@@ -378,6 +378,12 @@ def prepare_lora_cache(
|
|
| 378 |
try:
|
| 379 |
progress(0.35, desc="Building fused CPU transformer")
|
| 380 |
tmp_ledger = ledger.with_loras(tuple(loras_for_builder))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 381 |
|
| 382 |
orig_tmp_target = getattr(tmp_ledger, "_target_device", None)
|
| 383 |
orig_tmp_device = getattr(tmp_ledger, "device", None)
|
|
|
|
| 378 |
try:
|
| 379 |
progress(0.35, desc="Building fused CPU transformer")
|
| 380 |
tmp_ledger = ledger.with_loras(tuple(loras_for_builder))
|
| 381 |
+
# --- FIX: force tmp_ledger to use already downloaded checkpoints ---
|
| 382 |
+
tmp_ledger._distilled_checkpoint_path = checkpoint_path
|
| 383 |
+
tmp_ledger._spatial_upsampler_path = spatial_upsampler_path
|
| 384 |
+
tmp_ledger._gemma_root = gemma_root
|
| 385 |
+
# ---------------------------------------------------------------
|
| 386 |
+
assert os.path.exists(checkpoint_path), f"Checkpoint missing: {checkpoint_path}"
|
| 387 |
|
| 388 |
orig_tmp_target = getattr(tmp_ledger, "_target_device", None)
|
| 389 |
orig_tmp_device = getattr(tmp_ledger, "device", None)
|