Update README.md
#20
by
arteivissa
- opened
README.md
CHANGED
@@ -50,6 +50,15 @@ A .json file contains "caption" field with a text prompt.
|
|
50 |
python3 demo_lora_inference.py \
|
51 |
--checkpoint lora.safetensors \
|
52 |
--prompt " handsome girl in a suit covered with bold tattoos and holding a pistol. Animatrix illustration style, fantasy style, natural photo cinematic"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
```
|
54 |
|
55 |
![Example Picture 1](https://github.com/XLabs-AI/x-flux/blob/main/assets/readme/examples/picture-0-rev1.png?raw=true)
|
|
|
50 |
python3 demo_lora_inference.py \
|
51 |
--checkpoint lora.safetensors \
|
52 |
--prompt " handsome girl in a suit covered with bold tattoos and holding a pistol. Animatrix illustration style, fantasy style, natural photo cinematic"
|
53 |
+
# Cargar el LoRA con un peso ajustable
|
54 |
+
def load_lora_weights(pipe, lora_path, scale=0.75):
|
55 |
+
lora_weights = torch.load(lora_path)
|
56 |
+
# Aplicar el peso del LoRA (influencia)
|
57 |
+
pipe.unet.load_attn_procs(lora_weights, scale=scale)
|
58 |
+
|
59 |
+
# Generar la imagen con LoRA ajustado
|
60 |
+
image = pipe(prompt, num_inference_steps=50).images[0]
|
61 |
+
image.show()
|
62 |
```
|
63 |
|
64 |
![Example Picture 1](https://github.com/XLabs-AI/x-flux/blob/main/assets/readme/examples/picture-0-rev1.png?raw=true)
|