File size: 413 Bytes
45131ff |
1 2 3 4 5 6 7 8 9 10 |
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}") |