Spaces:

Skywork
/

UniPic2-Metaquery

Running on Zero

yichenchenchen commited on 5 days ago

Commit

00ad5ed

verified ·

1 Parent(s): 724b3a1

Update inferencer.py

Files changed (1) hide show

inferencer.py CHANGED Viewed

@@ -26,7 +26,7 @@ class UniPicV2Inferencer:
         self.quant = quant
         self.image_size = image_size
         self.default_negative_prompt = default_negative_prompt
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.pipeline = None#self._init_pipeline()
     def _init_pipeline(self) -> StableDiffusion3KontextPipeline:

         self.quant = quant
         self.image_size = image_size
         self.default_negative_prompt = default_negative_prompt
+        self.device = torch.device("cuda")
         self.pipeline = None#self._init_pipeline()
     def _init_pipeline(self) -> StableDiffusion3KontextPipeline: