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