Files changed (1) hide show
  1. README.md +9 -0
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)