mert_star / README.md
codermert's picture
Update README.md
35c531f verified
|
raw
history blame
333 Bytes
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]