Update README.md
Browse files
README.md
CHANGED
|
@@ -33,7 +33,7 @@ import torch
|
|
| 33 |
from transformers import AutoModel
|
| 34 |
|
| 35 |
model = AutoModel.from_pretrained(
|
| 36 |
-
"ControlNet/
|
| 37 |
trust_remote_code=True
|
| 38 |
)
|
| 39 |
tensor = torch.rand([1, 3, 16, 224, 224]) # (B, C, T, H, W)
|
|
|
|
| 33 |
from transformers import AutoModel
|
| 34 |
|
| 35 |
model = AutoModel.from_pretrained(
|
| 36 |
+
"ControlNet/marlin_vit_large_ytf", # or other variants
|
| 37 |
trust_remote_code=True
|
| 38 |
)
|
| 39 |
tensor = torch.rand([1, 3, 16, 224, 224]) # (B, C, T, H, W)
|