Add library name and pipeline tag; link to the paper

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

README.md +6 -1

README.md CHANGED Viewed

@@ -1,9 +1,14 @@
 ---
 license: mit
 ---
 [![Discord](https://img.shields.io/discord/232596713892872193?logo=discord)](https://discord.gg/2JhHVh7CGu)
-A semi custom network trained from scratch for 799 epochs based on [Simpler Diffusion (SiD2)](https://arxiv.org/abs/2410.19324v1)
 [Modeling](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/models/uvit.py) || [Training](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/train.py)

 ---
 license: mit
+library_name: diffusers
+pipeline_tag: image-to-image
 ---
 [![Discord](https://img.shields.io/discord/232596713892872193?logo=discord)](https://discord.gg/2JhHVh7CGu)
+A semi custom network trained from scratch for 799 epochs with tensor product attention. This repository contains the model
+described in [Tensor Product Attention Is All You Need](https://huggingface.co/papers/2501.06425).
+Github repository: https://github.com/tensorgi/T6
 [Modeling](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/models/uvit.py) || [Training](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/train.py)