Update README.md

README.md

@@ -10,17 +10,17 @@ tags:
 - text-to-video
 - video
 - video-generation
-base_model: "Wan-AI/Wan2.1-T2V-14B-Diffusers"
+base_model: "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
 pipeline_tag: text-to-video
-# widget:
-#   - text: >-
-#       prompt
-#     output:
-#       url: https://...
-instance_prompt: CYBREAL
+widget:
+  - text: >-
+      in the style of CYB77, driving a car very fast through a city at night
+    output:
+      url: https://replicate.delivery/xezq/asl3HD14zJpjCV3WAefO5cMDOqyxKG2ZWIfN1dD9DRby4V2oA/R8_Wan_00001.mp4
+instance_prompt: CYB77
 ---
 
-# Wan 14B Cyberpunk Realistic
+# Wan 14B Cyberpunk Realistic (image-to-video)
 
 <Gallery />
 
@@ -35,7 +35,7 @@ It was trained on [Replicate](https://replicate.com/) using AI toolkit: https://
 
 ## Trigger words
 
-You should use `CYBREAL` to trigger the video generation.
+You should use `CYB77` or "In the style ocf CYB77" to trigger the video generation.
 
 
 ## Use this LoRA
@@ -51,8 +51,9 @@ Replicate has a collection of Wan2.1 models that are optimised for speed and cos
 import replicate
 
 input = {
-    "prompt": "CYBREAL",
-    "lora_url": "https://huggingface.co/fofr/wan-14b-cyberpunk-realistic/resolve/main/wan2.1-14b-cybreal-lora.safetensors"
+    "prompt": "CYB77",
+    "image": "https://replicate.delivery/xezq/4BGR8w4ELvJfcqIG4KIF0Kr82JxtfQCM2xVAIUieSeifmTZjC/output_frame.jpg",
+    "lora_url": "https://huggingface.co/fofr/wan-14b-cyberpunk-realistic/resolve/main/wan-14b-i2v-cyberpunk-realistic-lora.safetensors"
 }
 
 output = replicate.run(
@@ -88,7 +89,7 @@ pipe.load_lora_weights("fofr/wan-14b-cyberpunk-realistic")
 
 pipe.enable_model_cpu_offload() #for low-vram environments
 
-prompt = "CYBREAL"
+prompt = "CYB77"
 negative_prompt = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards"
 
 output = pipe(
@@ -105,7 +106,7 @@ export_to_video(output, "output.mp4", fps=16)
 
 ## Training details
 
-- Steps: 2000
+- Steps: 5000
 - Learning rate: 0.0001
 - LoRA rank: 32