Update README.md
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ You should use `mert` to trigger the image generation.
|
|
30 |
import torch
|
31 |
from diffusers import FluxPipeline
|
32 |
|
33 |
-
pipe = FluxPipeline.from_pretrained("
|
34 |
pipe.load_lora_weights('codermert/mert_flux', weight_name='flux_train_replicate.safetensors')
|
35 |
pipe.fuse_lora(lora_scale=0.9)
|
36 |
pipe.to("cuda")
|
|
|
30 |
import torch
|
31 |
from diffusers import FluxPipeline
|
32 |
|
33 |
+
pipe = FluxPipeline.from_pretrained("prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0", torch_dtype=torch.bfloat16)
|
34 |
pipe.load_lora_weights('codermert/mert_flux', weight_name='flux_train_replicate.safetensors')
|
35 |
pipe.fuse_lora(lora_scale=0.9)
|
36 |
pipe.to("cuda")
|