| import os | |
| from safetensors.torch import load_file | |
| ckpt_path = r"C:\path\to\your\flux1-dev.safetensors" # or wherever your checkpoint is | |
| print(f"File size: {os.path.getsize(ckpt_path) / 1e9:.2f} GB") | |
| state_dict = load_file(ckpt_path) | |
| print(f"Total keys in checkpoint: {len(state_dict)}") | |
| # Check which double_blocks are present | |
| blocks = sorted(set([k.split('.')[1] for k in state_dict.keys() if 'double_blocks' in k])) | |
| print(f"Blocks present: {blocks}") |