Update README.md

README.md

@@ -65,8 +65,38 @@ The model reconstructs the clothing item while preserving its style, texture, co
 Built with fal.ai.
 
 ### Usage
-- Try LoRA on [fal.ai Playground](https://fal.ai/models/fal-ai/flux-2-klein/9b/base/edit/lora/playground?share=033cae58-12b5-4fef-992b-d48ff6759e01)
+- Try the LoRA on [fal.ai Playground](https://fal.ai/models/fal-ai/flux-2-klein/9b/base/edit/lora/playground?share=033cae58-12b5-4fef-992b-d48ff6759e01)
 - Try the workflow (image-to-video) at [fal.ai Workflow](https://fal.ai/workflows/rizavelioglu/workflow-virtual-try-off?view=playground)
+- ComfyUI: download compatible weights [virtual-tryoff-lora_comfy.safetensors](https://huggingface.co/fal/virtual-tryoff-lora/resolve/main/virtual-tryoff-lora_comfy.safetensors)
+- Diffusers:
+  ```python
+  import torch
+  from diffusers import Flux2KleinPipeline
+  from PIL import Image
+  
+  pipeline = Flux2KleinPipeline.from_pretrained(
+    "black-forest-labs/FLUX.2-klein-base-9B", 
+    torch_dtype=torch.bfloat16, 
+    low_cpu_mem_usage=False
+  ).to("cuda")
+  pipeline.load_lora_weights(
+      "fal/virtual-tryoff-lora", 
+      weight_name="virtual-tryoff-lora_diffusers.safetensors", 
+      adapter_name="vtoff"
+  )
+  pipeline.set_adapters("vtoff", adapter_weights=1.0)
+  pipeline.fuse_lora(adapter_names=["vtoff"], lora_scale=1.0)
+  
+  image = pipeline(
+      image=Image.open("<your_image>.jpg"),
+      prompt="TRYOFF extract the full outfit over a white background, product photography style.  NO HUMAN VISIBLE (the garments maintain their 3D form like an invisible mannequin).",
+      height=1024,
+      width=768,
+      num_inference_steps=28,
+      guidance_scale=5.0,
+      generator=torch.Generator("cuda").manual_seed(42),
+  ).images[0]
+  ```
 
 ### Training
 Trained with [fal.ai trainer](https://fal.ai/models/fal-ai/flux-2-klein-9b-base-trainer/edit).