Function is_vae_in_checkpoint

diffusers/scripts/convert_sd3_to_diffusers.py:192–195 · view source on GitHub ↗

(original_state_dict)

Source from the content-addressed store, hash-verified

190
191
192	def is_vae_in_checkpoint(original_state_dict):
193	return ("first_stage_model.decoder.conv_in.weight" in original_state_dict) and (
194	"first_stage_model.encoder.conv_in.weight" in original_state_dict
195	)
196
197
198	def main(args):

mainFunction · 0.85

no outgoing calls

no test coverage detected