Update README.md

README.md

@@ -52,8 +52,25 @@ license_link: >-
 You should use `Linear red light` to trigger the image generation.
 
 
-## Download model
+## Inference
 
-Weights for this model are available in Safetensors format.
+```python
+import torch
+from diffusers import StableDiffusion3Pipeline # please install diffusers from the source
 
-[Download](/Shakker-Labs/SD3.5-LoRA-Linear-Red-Light/tree/main) them in the Files & versions tab.
+pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-diffusers", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("Shakker-Labs/SD3.5-LoRA-Linear-Red-Light", weight_name="SD35-lora-Linear-Red-Light.safetensors")
+pipe.fuse_lora(lora_scale=1.0)
+pipe.to("cuda")
+
+prompt = "a cat, Linear red light"
+negative_prompt = "(lowres, low quality, worst quality)"
+
+image = pipe(prompt=prompt,
+             negative_prompt=negative_prompt
+             num_inference_steps=24, 
+             guidance_scale=4.0,
+             width=960, height=1280,
+            ).images[0]
+image.save(f"toy_example.jpg")
+```