Skip to content

Force CLIP text encoder onto CUDA if availabe to avoid CPU NaN outputs in Jetson Xavier#125

Open
mqcmd196 wants to merge 1 commit into
facebookresearch:mainfrom
mqcmd196:jetson-avoid-nan
Open

Force CLIP text encoder onto CUDA if availabe to avoid CPU NaN outputs in Jetson Xavier#125
mqcmd196 wants to merge 1 commit into
facebookresearch:mainfrom
mqcmd196:jetson-avoid-nan