Add pipeline tag and link to code

README.md

@@ -1,14 +1,15 @@
 ---
-language: en
-license: apache-2.0
-library_name: transformers
-tags:
-  - tptt
-  - peft
-  - trust_remote_code
 base_model: allenai/OLMo-1B-hf
 datasets:
 - yahma/alpaca-cleaned
+language: en
+library_name: transformers
+license: apache-2.0
+pipeline_tag: text-generation
+tags:
+- tptt
+- peft
+- trust_remote_code
 ---
 
 # Titans-OLMo-1B-hf
@@ -17,6 +18,7 @@ Titanesque version of `allenai/OLMo-1B-hf` with parallel linearized attention (T
 
 The model was presented in the paper [TPTT](https://huggingface.co/papers/2506.17671).
 
+For code, see https://github.com/fabienfrfr/tptt
 
 ## Model Details
 
@@ -70,5 +72,4 @@ print(tokenizer.decode(outputs, skip_special_tokens=True))
 
 If you use TPTT in your academic work, please cite [Furfaro](https://huggingface.co/ffurfaro). For questions or support, please open an issue on the [GitHub repository](https://github.com/fabienfrfr/tptt) or contact the maintainer.
 
-
 ---