Update app.py

app.py

@@ -190,9 +190,10 @@ def generate(
         state_file = f"rv_L_{segment-1}_{seed}.pt"
         state = torch.load(state_file, weights_only=False)
         # # TEST
+        #seed = state["seed"]
+        generator = torch.Generator(device='cuda').manual_seed(seed)
         latents = state["intermediate_latents"].to("cuda") #, dtype=torch.bfloat16)
         guidance_scale = state["guidance_scale"]
-        seed = state["seed"]
         all_timesteps_cpu = state["all_timesteps"]
         height = state["height"]
         width = state["width"]
@@ -200,7 +201,6 @@ def generate(
         timesteps_split_np = np.array_split(all_timesteps_cpu.numpy(), 2)
         timesteps_split_for_state = [chunk for chunk in timesteps_split_np]
         segment_timesteps = torch.from_numpy(timesteps_split_np[segment - 1]).to("cuda")
-        generator = torch.Generator(device='cuda').manual_seed(seed)
         prompt_embeds = state["prompt_embeds"].to("cuda", dtype=torch.bfloat16)
         negative_prompt_embeds = state["negative_prompt_embeds"].to("cuda", dtype=torch.bfloat16)
         pooled_prompt_embeds = state["pooled_prompt_embeds"].to("cuda", dtype=torch.bfloat16)