MCPcopy Create free account
hub / github.com/tdrussell/diffusion-pipe / empty_cuda_cache

Function empty_cuda_cache

utils/common.py:54–56  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

52
53
54def empty_cuda_cache():
55 gc.collect()
56 torch.cuda.empty_cache()
57
58
59@contextmanager

Callers 3

evaluateFunction · 0.90
train.pyFile · 0.90
load_diffusion_modelMethod · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected