DamarJati commited on
Commit
9c4729d
1 Parent(s): 364bea8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ torch.backends.cuda.matmul.allow_tf32 = False
13
  base_model = "black-forest-labs/FLUX.1-dev"
14
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
15
 
16
- pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
17
 
18
  lora_repo = "Shakker-Labs/AWPortraitCN"
19
  trigger_word = "" # Leave trigger_word blank if not used.
 
13
  base_model = "black-forest-labs/FLUX.1-dev"
14
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
15
 
16
+ # pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
17
 
18
  lora_repo = "Shakker-Labs/AWPortraitCN"
19
  trigger_word = "" # Leave trigger_word blank if not used.