Instructions to use Blackroot/SimpleDiffusion-TensorProductAttentionRope with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Blackroot/SimpleDiffusion-TensorProductAttentionRope with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Blackroot/SimpleDiffusion-TensorProductAttentionRope", dtype=torch.bfloat16, device_map="cuda") prompt = "Turn this cat into a dog" input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png") image = pipe(image=input_image, prompt=prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -7,8 +7,7 @@ pipeline_tag: image-to-image
|
|
| 7 |
[](https://discord.gg/2JhHVh7CGu)
|
| 8 |
|
| 9 |
A semi custom network trained from scratch for 799 epochs with tensor product attention. This repository contains the attention mechanism
|
| 10 |
-
described in [Tensor Product Attention Is All You Need](https://huggingface.co/papers/2501.06425) and the modeling is based on [Simpler Diffusion](https://arxiv.org/abs/2410.19324)
|
| 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 |
|
|
|
|
| 7 |
[](https://discord.gg/2JhHVh7CGu)
|
| 8 |
|
| 9 |
A semi custom network trained from scratch for 799 epochs with tensor product attention. This repository contains the attention mechanism
|
| 10 |
+
described in [Tensor Product Attention Is All You Need](https://huggingface.co/papers/2501.06425) [Tensor Product Attn Github](https://github.com/tensorgi/T6) and the modeling is based on [Simpler Diffusion](https://arxiv.org/abs/2410.19324)
|
|
|
|
| 11 |
|
| 12 |
[Modeling](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/models/uvit.py) || [Training](https://huggingface.co/Blackroot/SimpleDiffusion-TensorProductAttentionRope/blob/main/train.py)
|
| 13 |
|