CumfyUI Ksampler issue with Sana model

[faustoserone]

Hi,
using your's example workflow, give me this error:

KSampler
Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat1 in method wrapper_CUDA_addmm)

This happens if i launch comfy with the command "run_nvidia_gpu.bat".
With the command "run_cpu.bat" it works.
Changing the devices in Gemma loader.

Thanks in advance