How could I change to GPU model?

[vtchao]

I want to ask in GitHub when I use comfyUI to run the model, the model always runs in CPU; how could I change to GPU?

VAE load device: cuda:0, offload device: cpu, dtype: torch.bfloat16
model weight dtype torch.bfloat16, manual cast: None
model_type FLUX
CLIP/text encoder model load device: cuda:0, offload device: cpu, current: cpu, dtype: torch.float16