8 lines
332 B
Python
8 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))
|