KingNish commited on
Commit
b1301a2
1 Parent(s): 3417772

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -24,6 +24,10 @@ pipe = FluxWithCFGPipeline.from_pretrained(
24
  )
25
  pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
26
  pipe.to("cuda")
 
 
 
 
27
 
28
  torch.cuda.empty_cache()
29
 
 
24
  )
25
  pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
26
  pipe.to("cuda")
27
+ pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v2.3.safetensors', adapter_name="better")
28
+ pipe.set_adapters(["better"], adapter_weights=[1.0])
29
+ pipe.fuse_lora(adapter_name=["better"], lora_scale=1.0)
30
+ pipe.unload_lora_weights()
31
 
32
  torch.cuda.empty_cache()
33