dagloop5 commited on
Commit
46545b0
·
verified ·
1 Parent(s): 27694a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -166,11 +166,11 @@ def load_models() -> str | None:
166
  path = hf_hub_download(LORA_REPO, filename)
167
  with safe_open(path, framework="pt") as handle:
168
  keys = sorted(handle.keys())
169
- print(f"[lora-debug] {filename}: {len(keys)} keys", flush=True)
170
- for k in keys[:40]:
171
- print(f"[lora-debug] {k} {tuple(handle.get_slice(k).get_shape())}", flush=True)
172
- if len(keys) > 40:
173
- print(f"[lora-debug] ... and {len(keys) - 40} more", flush=True)
174
  except Exception as error:
175
  print(f"[lora-debug] failed to inspect {filename}: {error}", flush=True)
176
 
 
166
  path = hf_hub_download(LORA_REPO, filename)
167
  with safe_open(path, framework="pt") as handle:
168
  keys = sorted(handle.keys())
169
+ print(f"[lora-debug] {filename}: {len(keys)} keys", flush=True)
170
+ for k in keys[:40]:
171
+ print(f"[lora-debug] {k} {tuple(handle.get_slice(k).get_shape())}", flush=True)
172
+ if len(keys) > 40:
173
+ print(f"[lora-debug] ... and {len(keys) - 40} more", flush=True)
174
  except Exception as error:
175
  print(f"[lora-debug] failed to inspect {filename}: {error}", flush=True)
176