ControlNet
/

marlin_vit_large_ytf

Feature Extraction

Model card Files Files and versions

ControlNet commited on Mar 30

Commit

7907fef

·

verified ·

1 Parent(s): d3cd69e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ import torch
 from transformers import AutoModel
 model = AutoModel.from_pretrained(
-    "ControlNet/marlin_vit_base_ytf",  # or other variants
     trust_remote_code=True
 )
 tensor = torch.rand([1, 3, 16, 224, 224])  # (B, C, T, H, W)

 from transformers import AutoModel
 model = AutoModel.from_pretrained(
+    "ControlNet/marlin_vit_large_ytf",  # or other variants
     trust_remote_code=True
 )
 tensor = torch.rand([1, 3, 16, 224, 224])  # (B, C, T, H, W)