SwiftVR-bf16 / README.md
dummy9996's picture
Update README.md
c268db7 verified
|
Raw
History Blame
598 Bytes
---
base_model:
- H-oliday/SwiftVR
---
<h2><center><strong>SwiftVR: Real-Time One-Step Generative Video Restoration</strong></center></h2>
<center>Now in BF16</center>
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)