Update inf.py

inf.py

@@ -20,7 +20,7 @@ class InferencePipeline:
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         if self.device.type == 'cpu':
             self.pipe = StableDiffusionXLPipeline.from_pretrained(
-                self.base_model_id, use_auth_token=self.hf_token, cache_dir='./cache')
+                self.base_model_id, use_auth_token=self.hf_token)
         else:
             self.pipe = StableDiffusionXLPipeline.from_pretrained(
                 self.base_model_id,
@@ -111,6 +111,8 @@ class InferencePipeline:
         if not torch.cuda.is_available():
             raise gr.Error('CUDA is not available.')
 
+        print(f"running with {self.device}, {self.pipe.device}")
+
         self.load_pipe(content_lora_model_id, style_lora_model_id, content_alpha, style_alpha)
 
         generator = torch.Generator().manual_seed(seed)