Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ tags:
|
|
| 9 |
- diffusers
|
| 10 |
- lora
|
| 11 |
- replicate
|
| 12 |
-
base_model: "black-forest-labs/FLUX.1-
|
| 13 |
pipeline_tag: text-to-image
|
| 14 |
# widget:
|
| 15 |
# - text: >-
|
|
@@ -38,8 +38,7 @@ You should use `IRNMN` to trigger the image generation.
|
|
| 38 |
from diffusers import AutoPipelineForText2Image
|
| 39 |
import torch
|
| 40 |
|
| 41 |
-
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-
|
| 42 |
-
pipeline.set_options(output_format='webp', output_quality=100, disable_safety_checker=False, go_fast=True, megapixels=1)
|
| 43 |
pipeline.load_lora_weights('miike-ai/merlin-ironman', weight_name='lora.safetensors')
|
| 44 |
image = pipeline('your prompt').images[0]
|
| 45 |
```
|
|
|
|
| 9 |
- diffusers
|
| 10 |
- lora
|
| 11 |
- replicate
|
| 12 |
+
base_model: "black-forest-labs/FLUX.1-schnell"
|
| 13 |
pipeline_tag: text-to-image
|
| 14 |
# widget:
|
| 15 |
# - text: >-
|
|
|
|
| 38 |
from diffusers import AutoPipelineForText2Image
|
| 39 |
import torch
|
| 40 |
|
| 41 |
+
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
|
|
|
|
| 42 |
pipeline.load_lora_weights('miike-ai/merlin-ironman', weight_name='lora.safetensors')
|
| 43 |
image = pipeline('your prompt').images[0]
|
| 44 |
```
|