codermert commited on
Commit
35c531f
·
verified ·
1 Parent(s): 3519e50

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -5
README.md CHANGED
@@ -1,5 +1,7 @@
1
- ---
2
- license: other
3
- license_name: flux-1-dev-non-commercial-license
4
- license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
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
+ ```