Update README.md
Browse files
README.md
CHANGED
@@ -31,7 +31,7 @@ from diffusers import AutoPipelineForText2Image
|
|
31 |
import torch
|
32 |
|
33 |
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
|
34 |
-
pipeline.load_lora_weights('codermert/model_malika', weight_name='
|
35 |
image = pipeline('your prompt').images[0]
|
36 |
```
|
37 |
|
|
|
31 |
import torch
|
32 |
|
33 |
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
|
34 |
+
pipeline.load_lora_weights('codermert/model_malika', weight_name='pytorch_lora_weights.safetensors')
|
35 |
image = pipeline('your prompt').images[0]
|
36 |
```
|
37 |
|