Spaces:

frogleo
/

anime-ai-generator

Running on Zero

frogleo commited on Feb 4

Commit

58cfe29

1 Parent(s): 23063b3

补全丢失的方法

Files changed (2) hide show

__pycache__/utils.cpython-310.pyc CHANGED Viewed

Binary files a/__pycache__/utils.cpython-310.pyc and b/__pycache__/utils.cpython-310.pyc differ

utils.py CHANGED Viewed

@@ -93,3 +93,10 @@ def upscale(
     width = round(samples.shape[3] * scale_by)
     height = round(samples.shape[2] * scale_by)
     return common_upscale(samples, width, height, upscale_method)

     width = round(samples.shape[3] * scale_by)
     height = round(samples.shape[2] * scale_by)
     return common_upscale(samples, width, height, upscale_method)
+def free_memory() -> None:
+    """Free up GPU and system memory."""
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+        torch.cuda.ipc_collect()
+    gc.collect()