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))