Update app.py

app.py

@@ -49,7 +49,7 @@ def infer(
     progress=gr.Progress(track_tqdm=True),
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator("cuda").manual_seed(seed)
+    generator = torch.Generator().manual_seed(seed)
 
     # 使用 get_weighted_text_embeddings_sdxl 获取文本嵌入，不传递 device 参数
     if use_negative_prompt and negative_prompt:
@@ -79,14 +79,11 @@ def infer(
     image = pipe(
         prompt_embeds=prompt_embeds,
         negative_prompt_embeds=prompt_neg_embeds,
-        pooled_prompt_embeds          = pooled_prompt_embeds,
-        negative_pooled_prompt_embeds = negative_pooled_prompt_embeds,
         width=width,
         height=height,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
         generator=generator,
-        use_resolution_binning=use_resolution_binning,
     ).images[0]
     return image, seed