Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ 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('
|
49 |
image.save("my_image.png")
|
50 |
```
|
51 |
|
|
|
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 |
```
|
51 |
|