cleanup

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-18 16:54:33 +02:00 · 2024-10-22 20:31:35 -04:00
parent 4e4079e5e4
commit 142fddfe47
1 changed files with 1 additions and 0 deletions
@@ -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)