7 lines
332 B
Python
7 lines
332 B
Python
from safetensors import safe_open
|
|
|
|
|
|
with safe_open("/models/minimax_h3_fl2va_pruned_nvfp4.safetensors", framework="pt", device="cpu") as checkpoint:
|
|
for name in checkpoint.keys():
|
|
if name.startswith("condition_proj") or name.startswith("token_refiner"):
|
|
print(name, tuple(checkpoint.get_tensor(name).shape))
|