Update README.md
Browse files
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='
|
| 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 |
```
|