Method to

modules/onnx_impl/__init__.py:149–152 · view source on GitHub ↗

(self, *args, **kwargs)

Source from the content-addressed store, hash-verified

147	]
148
149	def to(self, args, *kwargs):
150	self.pipeline.vae_encoder = self.pipeline.vae_encoder.to(args, *kwargs)
151	self.pipeline.vae_decoder = self.pipeline.vae_decoder.to(args, *kwargs)
152	return self
153
154
155	def check_parameters_changed(p, refiner_enabled: bool):

create_latentsFunction · 0.45

full_vqgan_decodeFunction · 0.45

full_vae_decodeFunction · 0.45

full_vae_encodeFunction · 0.45

vae_encodeFunction · 0.45

cholesky_exFunction · 0.45

choleskyFunction · 0.45

_assign_custom_scheduleFunction · 0.45

_unipc_set_timesteps_device_fixFunction · 0.45

scale_noiseFunction · 0.45

reconstruct_multicond_batchFunction · 0.45

images_tensor_to_samplesFunction · 0.45

no outgoing calls

test_fp16Function · 0.36

test_bf16Function · 0.36

test_solverFunction · 0.36

test_load_end_to_end_with_bfl_prefix_no_converterFunction · 0.36

mainFunction · 0.36