Update inf.py

inf.py

@@ -113,9 +113,10 @@ class InferencePipeline:
 
         self.load_pipe(content_lora_model_id, style_lora_model_id, content_alpha, style_alpha)
         
-        self.pipe.to(self.device)
+        self.pipe.to("cuda")
 
-        generator = torch.Generator().manual_seed(seed)
+        generator = torch.Generator(device="cuda").manual_seed(seed)
+        print(self.pipe.device)
         out = self.pipe(
             prompt,
             num_inference_steps=n_steps,