MCPcopy Create free account
hub / github.com/CompVis/zigma / ckpt_wrapper

Method ckpt_wrapper

model_zigma.py:904–909  ·  view source on GitHub ↗
(self, module)

Source from the content-addressed store, hash-verified

902 return imgs
903
904 def ckpt_wrapper(self, module):
905 def ckpt_forward(*inputs):
906 outputs = module(*inputs)
907 return outputs
908
909 return ckpt_forward
910
911 def forward(
912 self,

Callers 1

forwardMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected