Update README.md
Browse files
README.md
CHANGED
@@ -31,7 +31,7 @@ import torch
|
|
31 |
from diffusers import FluxPipeline
|
32 |
|
33 |
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
|
34 |
-
pipe.load_lora_weights('codermert/
|
35 |
pipe.fuse_lora(lora_scale=0.9)
|
36 |
pipe.to("cuda")
|
37 |
|
|
|
31 |
from diffusers import FluxPipeline
|
32 |
|
33 |
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
|
34 |
+
pipe.load_lora_weights('codermert/ayd_flux', weight_name='flux_train_replicate.safetensors')
|
35 |
pipe.fuse_lora(lora_scale=0.9)
|
36 |
pipe.to("cuda")
|
37 |
|