mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 22:35:09 +00:00
10 lines
130 B
Python
10 lines
130 B
Python
import gc
|
|
|
|
import torch
|
|
|
|
|
|
def clean_memory():
|
|
if torch.cuda.is_available():
|
|
torch.cuda.empty_cache()
|
|
gc.collect()
|