Spaces:

stepfun-ai
/

Step1X-Edit

Running on Zero

listen2you003 commited on 3 days ago

Commit

723b769

1 Parent(s): b812142

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -317,7 +317,7 @@ class ImageGenerator:
         ref_images_raw = self.load_image(ref_images_raw)
         ref_images_raw = ref_images_raw.to(self.device)
-        print(f'self.ae, self.dit device: {self.ae.device}, {self.dit.device}')
         ref_images = self.ae.encode(ref_images_raw.to(self.device) * 2 - 1)
         seed = int(seed)

         ref_images_raw = self.load_image(ref_images_raw)
         ref_images_raw = ref_images_raw.to(self.device)
+        # print(f'self.ae, self.dit device: {self.ae.device}, {self.dit.device}')
         ref_images = self.ae.encode(ref_images_raw.to(self.device) * 2 - 1)
         seed = int(seed)