Update app.py

app.py

@@ -13,6 +13,8 @@ torch.backends.cuda.matmul.allow_tf32 = False
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 
+pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
+
 lora_repo = "Shakker-Labs/AWPortraitCN"
 trigger_word = ""  # Leave trigger_word blank if not used.
 pipe.load_lora_weights(lora_repo, low_cpu_mem_usage=True)