Update inference_manager.py

inference_manager.py

@@ -508,7 +508,7 @@ class ModelManager:
         seed = seed or int(randomize_seed_fn(seed, randomize_seed))
         p = remove_child_related_content(p)
         prompt_str = cfg.get("prompt", "{prompt}").replace("{prompt}", p)
-        generator = torch.Generator(pipe.device).manual_seed(seed)
+        generator = torch.Generator(model.base_model_pipeline.device).manual_seed(seed)
         print(f"generate: p={p}, np={np}, steps={steps}, guidance_scale={guidance_scale}, size={width},{height}, seed={seed}")
         images = ip_model.generate(
             prompt=prompt_str,