Update app.py

app.py

@@ -28,9 +28,8 @@ pipe = FluxKontextPipeline.from_pretrained(
     torch_dtype=dtype
 ).to(device)
 
-# Set scheduler to match ComfyUI (Euler with simple scheduler)
-pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
-print("📅 Using Euler scheduler (matching ComfyUI)")
+# Keep the default scheduler that works with FluxKontextPipeline
+print(f"📅 Using default scheduler: {pipe.scheduler.__class__.__name__}")
 
 pipe.load_lora_weights(
     "DoozyWo/Kontext_avatar_LoRA", 
@@ -103,13 +102,12 @@ def transform_to_avatar(input_image, custom_prompt="", guidance_scale=1.0, num_i
     
     print(f"Avatar transformation prompt: {prompt_with_template}")
     
-    # Match ComfyUI parameters exactly: cfg=1, steps=25, sampler=euler, scheduler=simple
+    # Match ComfyUI parameters: cfg=1, steps=25
+    # Let the pipeline handle dimensions automatically
     image = pipe(
         image=scaled_image, 
         prompt=prompt_with_template,
         guidance_scale=guidance_scale,  # CFG=1 from ComfyUI
-        width=scaled_image.size[0],
-        height=scaled_image.size[1],
         num_inference_steps=num_inference_steps,
         generator=torch.Generator(device=pipe.device).manual_seed(seed),
     ).images[0]