Class Upsample

wan/modules/vae.py:57–63 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

55
56
57	class Upsample(nn.Upsample):
58
59	def forward(self, x):
60	"""
61	Fix bfloat16 support for nearest neighbor interpolation.
62	"""
63	return super().forward(x.float()).type_as(x)
64
65
66	class Resample(nn.Module):

__init__Method · 0.85

no outgoing calls

no test coverage detected