nielsr HF Staff commited on
Commit
2098f39
·
verified ·
1 Parent(s): 475b384

Add pipeline tag

Browse files

This 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.

Files changed (1) hide show
  1. README.md +3 -2
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
+ ```