dribnet / pixray

neural image generation
Other
402 stars 53 forks source link

Changes to support Torch 1.10 for CUDA 11 #34

Closed bfrasure closed 2 years ago

bfrasure commented 2 years ago

Turned off jit for torch 1.8 on up. jit does not work with torch 1.10 either so this is no loss.

dribnet commented 2 years ago

Thanks @bfrasure for this fix.