CUDA Error: unknown error when generating an image (model trained with --taming flag)

[vladislabv]

Hello everyone!

I am facing an issue while trying to generate an image, after a model was successfully trained. The training have done with a docker image provided in the repository. After training, I also get True by testing torch.cuda in the python compiler, so I do not get where the error comes from.
Here is the error traceback:

Behind docker I am using Windows 10, Nvidia GeForce 1050 as GPU, Intel(R) Core(TM) i5-8300H CPU @ 2.30GHz 2.30 GHz as CPU and 8 GB RAM.
Here is the output of the cuda being used on docker:

As well as drivers being used on the PC:

I would appreciate any help, if someone have experienced the same problem!