dagloop5 commited on
Commit
d7cff3a
·
verified ·
1 Parent(s): 6091868

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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)