Method get_input

sat/diffusion_video.py:173–174 · view source on GitHub ↗

(self, batch)

Source from the content-addressed store, hash-verified

171	return loss, loss_dict
172
173	def get_input(self, batch):
174	return batch[self.input_key].to(self.dtype)
175
176	@torch.no_grad()
177	def decode_first_stage(self, z):

shared_stepMethod · 0.95

log_videoMethod · 0.95

no outgoing calls

no test coverage detected