nielsr HF staff commited on
Commit
93ce4dc
·
verified ·
1 Parent(s): 4e1336c

Add library name and pipeline tag; link to the paper

Browse files

This PR improves the model card by adding metadata such as `library_name` and `pipeline_tag`. It adds a description about the model in relation to the paper [Tensor Product Attention Is All You Need](https://huggingface.co/papers/2501.06425).

Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -1,9 +1,14 @@
1
  ---
2
  license: mit
 
 
3
  ---
 
4
  [![Discord](https://img.shields.io/discord/232596713892872193?logo=discord)](https://discord.gg/2JhHVh7CGu)
5
 
6
- A semi custom network trained from scratch for 799 epochs based on [Simpler Diffusion (SiD2)](https://arxiv.org/abs/2410.19324v1)
 
 
7
 
8
  [Modeling](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/models/uvit.py) || [Training](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/train.py)
9
 
 
1
  ---
2
  license: mit
3
+ library_name: diffusers
4
+ pipeline_tag: image-to-image
5
  ---
6
+
7
  [![Discord](https://img.shields.io/discord/232596713892872193?logo=discord)](https://discord.gg/2JhHVh7CGu)
8
 
9
+ A semi custom network trained from scratch for 799 epochs with tensor product attention. This repository contains the model
10
+ described in [Tensor Product Attention Is All You Need](https://huggingface.co/papers/2501.06425).
11
+ Github repository: https://github.com/tensorgi/T6
12
 
13
  [Modeling](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/models/uvit.py) || [Training](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/train.py)
14