/src
Running Stage 1: Overall restoration
detection.py:25: DeprecationWarning: BICUBIC is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.BICUBIC instead.
def data_transforms(img, full_size, method=Image.BICUBIC):
initializing the dataloader
model weights loaded
directory of testing image: /src/input/cog_temp
processing tmp_50t16tj1971 Midgets-Solo.jpg
test.py:18: DeprecationWarning: BILINEAR is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.BILINEAR instead.
def data_transforms(img, method=Image.BILINEAR, scale=False):
You are using NL + Res
Now you are processing tmp_50t16tj1971 Midgets-Solo.png
Skip tmp_50t16tj1971 Midgets-Solo.png due to an error:
CUDA out of memory. Tried to allocate 32.35 GiB (GPU 0; 14.58 GiB total capacity; 1.13 GiB already allocated; 8.00 GiB free; 5.40 GiB reserved in total by PyTorch)
Finish Stage 1 ...
Running Stage 2: Face Detection
Finish Stage 2 ...
Running Stage 3: Face Enhancement
The main GPU is
0
dataset [FaceTestDataset] of size 0 was created
The size of the latent vector size is [16,16]
Network [SPADEGenerator] was created. Total number of parameters: 92.1 million. To see the architecture, do print(network).
hi :)
Finish Stage 3 ...
Running Stage 4: Blending
Finish Stage 4 ...
All the processing is done. Please check the results.
Traceback (most recent call last):
File "/root/.pyenv/versions/3.8.14/lib/python3.8/site-packages/cog/server/worker.py", line 221, in _predict
result = predict(**payload)
File "predict.py", line 177, in predict
final_output = os.listdir(os.path.join(self.output_folder, "final_output"))[0]
IndexError: list index out of range
/src Running Stage 1: Overall restoration detection.py:25: DeprecationWarning: BICUBIC is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.BICUBIC instead. def data_transforms(img, full_size, method=Image.BICUBIC): initializing the dataloader model weights loaded directory of testing image: /src/input/cog_temp processing tmp_50t16tj1971 Midgets-Solo.jpg test.py:18: DeprecationWarning: BILINEAR is deprecated and will be removed in Pillow 10 (2023-07-01). Use Resampling.BILINEAR instead. def data_transforms(img, method=Image.BILINEAR, scale=False): You are using NL + Res Now you are processing tmp_50t16tj1971 Midgets-Solo.png Skip tmp_50t16tj1971 Midgets-Solo.png due to an error: CUDA out of memory. Tried to allocate 32.35 GiB (GPU 0; 14.58 GiB total capacity; 1.13 GiB already allocated; 8.00 GiB free; 5.40 GiB reserved in total by PyTorch) Finish Stage 1 ... Running Stage 2: Face Detection Finish Stage 2 ... Running Stage 3: Face Enhancement The main GPU is 0 dataset [FaceTestDataset] of size 0 was created The size of the latent vector size is [16,16] Network [SPADEGenerator] was created. Total number of parameters: 92.1 million. To see the architecture, do print(network). hi :) Finish Stage 3 ... Running Stage 4: Blending Finish Stage 4 ... All the processing is done. Please check the results. Traceback (most recent call last): File "/root/.pyenv/versions/3.8.14/lib/python3.8/site-packages/cog/server/worker.py", line 221, in _predict result = predict(**payload) File "predict.py", line 177, in predict final_output = os.listdir(os.path.join(self.output_folder, "final_output"))[0] IndexError: list index out of range
how do i correct this?