diff --git a/modules/model_omnigen.py b/modules/model_omnigen.py
index 9212193dd..64c99ddd7 100644
--- a/modules/model_omnigen.py
+++ b/modules/model_omnigen.py
@@ -13,6 +13,7 @@ def load_omnigen(checkpoint_info, diffusers_load_config={}): # pylint: disable=u
     # init
     pipe.device = devices.device
     pipe.dtype = devices.dtype
+    pipe.model.device = devices.device
     pipe.separate_cfg_infer = True
     pipe.use_kv_cache = False
     pipe.model.to(device=devices.device, dtype=devices.dtype)