The error points towards a device mismatch and it seems you haven’t pushed the parameters of the model to the device, so you might want to add gen.to('cuda').



 Merwa_Mehmood:

Is this the reason i am not getting fake images?


If this error was also raised before, the gen model shouldn’t have worked and I’m unsure where the initial output image is coming from.