| ```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] | |
| ``` |