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