Update README.md
Browse files
README.md
CHANGED
|
@@ -105,7 +105,8 @@ pipeline = FluxPipeline.from_pretrained(
|
|
| 105 |
|
| 106 |
#pipeline.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Comment out if you have enough GPU VRAM
|
| 107 |
|
| 108 |
-
pipeline.load_lora_weights(repo_id,
|
|
|
|
| 109 |
pipeline=pipeline.to('cuda')
|
| 110 |
```
|
| 111 |
Image generation - Example #1:
|
|
|
|
| 105 |
|
| 106 |
#pipeline.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Comment out if you have enough GPU VRAM
|
| 107 |
|
| 108 |
+
pipeline.load_lora_weights(repo_id, #weight_name=f'XXX.safetensors'
|
| 109 |
+
)
|
| 110 |
pipeline=pipeline.to('cuda')
|
| 111 |
```
|
| 112 |
Image generation - Example #1:
|