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