modified: app.py

app.py

@@ -185,6 +185,7 @@ class FluxEditor:
             torch.cuda.empty_cache()
             self.model = self.model.to(self.device)
 
+        self.model = self.model.to(self.device)
         # denoise initial noise
         info = {}
         info['feature'] = {}
@@ -419,14 +420,14 @@ class FluxEditor:
         #--------------------     6.1 prepare output folder      ----------------------#
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)
-            idx = 1
+            idx = 0
         #--------------------    6.2 editing round      ----------------------#           
         else:
             fns = [fn for fn in os.listdir(self.output_dir)]
             if len(fns) > 0:
                 idx = max(int(fn.split("_")[1]) for fn in fns) + 1
             else:
-                idx = 1
+                idx = 0
         formatted_idx = str(idx).zfill(4) # Format as a 4-digit string
 
         #--------------------    6.3 output name      ----------------------#