Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,7 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
|
|
|
|
|
1 |
+
```py
|
2 |
+
from diffusers import AutoPipelineForText2Image
|
3 |
+
import torch
|
4 |
+
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
|
5 |
+
pipeline.load_lora_weights('codermert/mert_star', weight_name='flux_train_replicate.safetensors')
|
6 |
+
image = pipeline('your prompt').images[0]
|
7 |
+
```
|