Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 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 |
|