harsha19 commited on
Commit
08a7d77
·
verified ·
1 Parent(s): 7a123a4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -10,7 +10,7 @@ tags:
10
  - lora
11
  - replicate
12
  base_model:
13
- - multimodalart/FLUX.1-dev2pro-full
14
  pipeline_tag: text-to-image
15
  instance_prompt: rups
16
  ---
@@ -34,7 +34,7 @@ You should use `rups` to trigger the image generation.
34
  from diffusers import AutoPipelineForText2Image
35
  import torch
36
 
37
- pipeline = AutoPipelineForText2Image.from_pretrained('multimodalart/FLUX.1-dev2pro-full', torch_dtype=torch.float16).to('cuda')
38
  pipeline.load_lora_weights('harshasai-dev/rupss', weight_name='lora.safetensors')
39
  image = pipeline('your prompt').images[0]
40
  ```
 
10
  - lora
11
  - replicate
12
  base_model:
13
+ - black-forest-labs/FLUX.1-schnell
14
  pipeline_tag: text-to-image
15
  instance_prompt: rups
16
  ---
 
34
  from diffusers import AutoPipelineForText2Image
35
  import torch
36
 
37
+ pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-schnell', torch_dtype=torch.float16).to('cuda')
38
  pipeline.load_lora_weights('harshasai-dev/rupss', weight_name='lora.safetensors')
39
  image = pipeline('your prompt').images[0]
40
  ```