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/mert2_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("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.bfloat16)
|
34 |
pipe.load_lora_weights('codermert/mert2_flux', weight_name='flux_train_replicate.safetensors')
|
35 |
pipe.fuse_lora(lora_scale=0.9)
|
36 |
pipe.to("cuda")
|