Vijish commited on
Commit
fb729c9
·
verified ·
1 Parent(s): 5737c96

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -44,7 +44,7 @@ from diffusers import AutoPipelineForText2Image
44
  import torch
45
 
46
  pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
47
- pipeline.load_lora_weights('Vijish/vijishtest', weight_name='vijishtest')
48
  image = pipeline('a photo of pushpa').images[0]
49
  image.save("my_image.png")
50
  ```
 
44
  import torch
45
 
46
  pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
47
+ pipeline.load_lora_weights('Vijish/vijishtest', weight_name='flux_train_replicate.safetensors')
48
  image = pipeline('a photo of pushpa').images[0]
49
  image.save("my_image.png")
50
  ```