Update inf.py

inf.py

@@ -108,14 +108,12 @@ class InferencePipeline:
             guidance_scale: float,
             num_images_per_prompt: int = 1
     ) -> PIL.Image.Image:
-        
-        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         if not torch.cuda.is_available():
             raise gr.Error('CUDA is not available.')
 
         self.load_pipe(content_lora_model_id, style_lora_model_id, content_alpha, style_alpha)
 
-        generator = torch.Generator(device=self.device).manual_seed(seed)
+        generator = torch.Generator().manual_seed(seed)
         out = self.pipe(
             prompt,
             num_inference_steps=n_steps,