Model card: repo id

README.md

@@ -27,7 +27,7 @@ With `transformers` only:
 import torch
 from transformers import AutoModel
 
-model = AutoModel.from_pretrained("<this repo>", trust_remote_code=True).cuda().eval()
+model = AutoModel.from_pretrained("KBlueLeaf/TTVidT", trust_remote_code=True).cuda().eval()
 video = torch.rand(1, 8, 3, 256, 256, device="cuda") * 2 - 1   # [B, T, C, H, W], pixels in [-1, 1]
 with torch.no_grad(), torch.autocast("cuda", dtype=torch.float16):
     motion = model(video).motion_output                          # [B, T, 1, 768]
@@ -38,7 +38,7 @@ With the TT-VidT repository:
 ```python
 from ttvidt.hub import load_model
 
-model = load_model("<this repo>", device="cuda")
+model = load_model("KBlueLeaf/TTVidT", device="cuda")
 motion = model.encoder(video).motion_output
 ```