Add pipeline tag (#1)
Browse files- Add pipeline tag (25ab37723c2830fc1b1f82827c751b5eef74ab70)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
|
@@ -1,9 +1,11 @@
|
|
| 1 |
---
|
| 2 |
-
license: bsd-3-clause
|
| 3 |
language:
|
| 4 |
- en
|
|
|
|
| 5 |
tags:
|
| 6 |
- arxiv:2506.23151
|
|
|
|
|
|
|
| 7 |
---
|
| 8 |
|
| 9 |
# MEMFOF-Tartan-T-TSKH
|
|
@@ -32,4 +34,4 @@ model = MEMFOF.from_pretrained("egorchistov/MEMFOF-Tartan-T-TSKH").eval().to(dev
|
|
| 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 |
+
library_name: pytorch
|
| 9 |
---
|
| 10 |
|
| 11 |
# MEMFOF-Tartan-T-TSKH
|
|
|
|
| 34 |
with torch.inference_mode():
|
| 35 |
example_input = torch.randint(0, 256, [1, 3, 3, 1080, 1920], device=device) # [B=1, T=3, C=3, H=1080, W=1920]
|
| 36 |
backward_flow, forward_flow = model(example_input).unbind(dim=1) # [B=1, C=2, H=1080, W=1920]
|
| 37 |
+
```
|