File size: 333 Bytes
35c531f |
1 2 3 4 5 6 7 |
```py
from diffusers import AutoPipelineForText2Image
import torch
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
pipeline.load_lora_weights('codermert/mert_star', weight_name='flux_train_replicate.safetensors')
image = pipeline('your prompt').images[0]
``` |