Class Upsample

wan/models/wan_vae.py:61–67 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

__init__Method · 0.70

no outgoing calls

no test coverage detected