init

app.py

@@ -9,7 +9,7 @@ from pipeline_freescale import StableDiffusionXLPipeline
 from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
 
 @spaces.GPU(duration=120)
-def infer_gpu_part(pipe, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu):
+def infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu):
     pipe = pipe.to("cuda") 
     generator = torch.Generator(device='cuda')
     generator = generator.manual_seed(seed)
@@ -46,7 +46,7 @@ def infer(prompt, output_size, ddim_steps, guidance_scale, cosine_scale, seed, o
     pipe = StableDiffusionXLPipeline.from_pretrained(model_ckpt, torch_dtype=torch.float16)
 
     print('GPU starts')
-    result = infer_gpu_part(pipe, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu)
+    result = infer_gpu_part(pipe, seed, prompt, negative_prompt, ddim_steps, guidance_scale, resolutions_list, fast_mode, cosine_scale, disable_freeu)
     print('GPU ends')
 
     image = result.images[0]