| from safetensors import safe_open | |
| with safe_open("flux1-depth-dev_fp4_merged_model.safetensors", framework="pt") as f: | |
| keys = f.keys() | |
| has_vae = any('vae' in k or 'decoder' in k for k in keys) | |
| has_clip = any('text_encoder' in k or 'clip' in k for k in keys) | |
| has_unet = any('unet' in k or 'transformer' in k for k in keys) | |
| print(f"VAE: {has_vae}, CLIP: {has_clip}, UNet: {has_unet}") |