SwiftVR-bf16 / README.md
dummy9996's picture
Update README.md
c268db7 verified
|
Raw
History Blame
598 Bytes
metadata
base_model:
  - H-oliday/SwiftVR

SwiftVR: Real-Time One-Step Generative Video Restoration

Now in BF16

Original model is at https://huggingface.co/H-oliday/SwiftVR

Install:

git clone https://github.com/H-oliday/SwiftVR

cd SwiftVR

uv pip install -e .

hf auth login --token {token}

hf download dummy9996/SwiftVR-bf16 --local-dir checkpoints/

Run:

from swiftvr import SwiftVRPipeline

pipe = SwiftVRPipeline.from_pretrained("checkpoints/").to("cuda", dtype="bfloat16")

pipe.restore_video("before.mp4", "after.mp4", upscale=2)