diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-08 17:13:18 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-08 17:13:18 +0300 |
| commit | da8916f92649fc4d947cb46d9d8f8ea1621b2a59 (patch) | |
| tree | b3c6883f8c731b0e1291e61a16f16b12ed22e101 /extensions-builtin/ScuNET/scripts/scunet_model.py | |
| parent | e161b5a0259c870b9d01408d02c504c3281dbdb1 (diff) | |
added torch.mps.empty_cache() to torch_gc()
changed a bunch of places that use torch.cuda.empty_cache() to use torch_gc() instead
Diffstat (limited to 'extensions-builtin/ScuNET/scripts/scunet_model.py')
| -rw-r--r-- | extensions-builtin/ScuNET/scripts/scunet_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/extensions-builtin/ScuNET/scripts/scunet_model.py b/extensions-builtin/ScuNET/scripts/scunet_model.py index ffef26b2..167d2f64 100644 --- a/extensions-builtin/ScuNET/scripts/scunet_model.py +++ b/extensions-builtin/ScuNET/scripts/scunet_model.py @@ -85,7 +85,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler): def do_upscale(self, img: PIL.Image.Image, selected_file): - torch.cuda.empty_cache() + devices.torch_gc() try: model = self.load_model(selected_file) @@ -110,7 +110,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler): torch_output = torch_output[:, :h * 1, :w * 1] # remove padding, if any np_output: np.ndarray = torch_output.float().cpu().clamp_(0, 1).numpy() del torch_img, torch_output - torch.cuda.empty_cache() + devices.torch_gc() output = np_output.transpose((1, 2, 0)) # CHW to HWC output = output[:, :, ::-1] # BGR to RGB |