AUTOMATIC1111 / stable-diffusion-webui-pixelization

stable-diffusion-webui-pixelization
MIT License
550 stars 72 forks source link

Memory leak #11

Open DenkingOfficial opened 1 year ago

DenkingOfficial commented 1 year ago

After some pixelization processes it throws an OOM error


100%|████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  1.08it/s]
Total progress: 100%|████████████████████████████████████████████████████████████████████| 1/1 [00:01<00:00,  1.62s/it]
100%|████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:02<00:00,  2.13s/it]
Total progress: 100%|████████████████████████████████████████████████████████████████████| 1/1 [00:03<00:00,  3.70s/it]
Error completing request
Arguments: (0, <PIL.Image.Image image mode=RGB size=900x1200 at 0x28A3B904DC0>, None, '', '', True, 0, 4, 512, 512, True, 'None', 'None', 0, 0, 0, 0, True, True, 2) {}
Traceback (most recent call last):
  File "C:\Software\Other Software\sdautomatic111\modules\call_queue.py", line 56, in f
    res = list(func(*args, **kwargs))
  File "C:\Software\Other Software\sdautomatic111\modules\call_queue.py", line 37, in f
    res = func(*args, **kwargs)
  File "C:\Software\Other Software\sdautomatic111\modules\postprocessing.py", line 56, in run_postprocessing
    scripts.scripts_postproc.run(pp, args)
  File "C:\Software\Other Software\sdautomatic111\modules\scripts_postprocessing.py", line 130, in run
    script.process(pp, **process_args)
  File "C:\Software\Other Software\sdautomatic111\extensions\stable-diffusion-webui-pixelization\scripts\postprocessing_pixelization.py", line 209, in process
    images = self.model.G_A_net.module.RGBDec(feature, adain_params)
  File "C:\Software\Other Software\sdautomatic111\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
    return forward_call(*input, **kwargs)
  File "C:\Software\Other Software\sdautomatic111\extensions\stable-diffusion-webui-pixelization\pixelization\models\c2pGen.py", line 260, in forward
    x = self.conv_2(x)
  File "C:\Software\Other Software\sdautomatic111\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
    return forward_call(*input, **kwargs)
  File "C:\Software\Other Software\sdautomatic111\extensions\stable-diffusion-webui-pixelization\pixelization\models\basic_layer.py", line 251, in forward
    x = self.conv(self.pad(x))
  File "C:\Software\Other Software\sdautomatic111\venv\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
    return forward_call(*input, **kwargs)
  File "C:\Software\Other Software\sdautomatic111\extensions-builtin\Lora\lora.py", line 182, in lora_Conv2d_forward
    return lora_forward(self, input, torch.nn.Conv2d_forward_before_lora(self, input))
  File "C:\Software\Other Software\sdautomatic111\venv\lib\site-packages\torch\nn\modules\conv.py", line 457, in forward
    return self._conv_forward(input, self.weight, self.bias)
  File "C:\Software\Other Software\sdautomatic111\venv\lib\site-packages\torch\nn\modules\conv.py", line 453, in _conv_forward
    return F.conv2d(input, weight, bias, self.stride,
RuntimeError: CUDA out of memory. Tried to allocate 1.03 GiB (GPU 0; 8.00 GiB total capacity; 5.75 GiB already allocated; 0 bytes free; 6.59 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.  See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF```