Update README.md

README.md

@@ -1,7 +1,7 @@
 ---
 license: other
-license_name: flux-1-dev-non-commercial-license
-license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
+license_name: flux-1-schnell-non-commercial-license
+license_link: https://huggingface.co/black-forest-labs/FLUX.1-schnell/blob/main/LICENSE.md
 language:
 - en
 tags:
@@ -10,7 +10,7 @@ tags:
 - lora
 - replicate
 base_model:
-- black-forest-labs/FLUX.1-dev
+- black-forest-labs/FLUX.1-schnell
 pipeline_tag: text-to-image
 instance_prompt: SABA
 ---
@@ -32,7 +32,7 @@ You should use `SABA` to trigger the image generation.
 from diffusers import AutoPipelineForText2Image
 import torch
 
-pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
+pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-schnell', torch_dtype=torch.float16).to('cuda')
 pipeline.load_lora_weights('janannfndnd/SABA', weight_name='lora.safetensors')
 image = pipeline('your prompt').images[0]
 ```