modified: app.py

app.py

@@ -138,7 +138,7 @@ class FluxEditor:
     ):
 
         if seed is None:
-            g_seed = torch.Generator(device="cpu").seed()
+            g_seed = torch.Generator(device=torch.device("cpu")).seed()
         print(f"Generating '{prompt}' with seed {g_seed}")
         t0 = time.perf_counter()
 
@@ -283,7 +283,7 @@ class FluxEditor:
             init_image_2 = np.array(init_image_2_pil)  # Convert back to NumPy (if needed)
             init_image_2 = encode(init_image_2, self.device, self.ae)
 
-        rng = torch.Generator(device="cpu")
+        rng = torch.Generator(device=torch.device("cpu"))
         opts = SamplingOptions(
             source_prompt=source_prompt,
             target_prompt=target_prompt,
@@ -294,7 +294,7 @@ class FluxEditor:
             seed=seed,
         )
         if opts.seed is None:
-            opts.seed = torch.Generator(device="cpu").seed()
+            opts.seed = torch.Generator(device=torch.device("cpu")).seed()
         
         print(f"Editing with prompt:\n{opts.source_prompt}")
         t0 = time.perf_counter()