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 |
|