Add pipeline tag
Browse filesThis PR adds the `pipeline_tag` to the model card, which helps improve discoverability on the Hugging Face Hub by ensuring the model appears under the "image-to-image" filter: https://huggingface.co/models?pipeline_tag=image-to-image.
README.md
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
---
|
| 2 |
-
license: bsd-3-clause
|
| 3 |
language:
|
| 4 |
- en
|
|
|
|
| 5 |
tags:
|
| 6 |
- arxiv:2506.23151
|
|
|
|
| 7 |
---
|
| 8 |
|
| 9 |
# MEMFOF-Tartan-T
|
|
@@ -32,4 +33,4 @@ model = MEMFOF.from_pretrained("egorchistov/MEMFOF-Tartan-T").eval().to(device)
|
|
| 32 |
with torch.inference_mode():
|
| 33 |
example_input = torch.randint(0, 256, [1, 3, 3, 1080, 1920], device=device) # [B=1, T=3, C=3, H=1080, W=1920]
|
| 34 |
backward_flow, forward_flow = model(example_input).unbind(dim=1) # [B=1, C=2, H=1080, W=1920]
|
| 35 |
-
```
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
language:
|
| 3 |
- en
|
| 4 |
+
license: bsd-3-clause
|
| 5 |
tags:
|
| 6 |
- arxiv:2506.23151
|
| 7 |
+
pipeline_tag: image-to-image
|
| 8 |
---
|
| 9 |
|
| 10 |
# MEMFOF-Tartan-T
|
|
|
|
| 33 |
with torch.inference_mode():
|
| 34 |
example_input = torch.randint(0, 256, [1, 3, 3, 1080, 1920], device=device) # [B=1, T=3, C=3, H=1080, W=1920]
|
| 35 |
backward_flow, forward_flow = model(example_input).unbind(dim=1) # [B=1, C=2, H=1080, W=1920]
|
| 36 |
+
```
|