MCPcopy Index your code
hub / github.com/huggingface/diffusers / is_vae_in_checkpoint

Function is_vae_in_checkpoint

scripts/convert_sd3_to_diffusers.py:241–244  ·  view source on GitHub ↗
(original_state_dict)

Source from the content-addressed store, hash-verified

239
240
241def is_vae_in_checkpoint(original_state_dict):
242 return ("first_stage_model.decoder.conv_in.weight" in original_state_dict) and (
243 "first_stage_model.encoder.conv_in.weight" in original_state_dict
244 )
245
246
247def get_attn2_layers(state_dict):

Callers 1

mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…