Downloads · 30 days
14
6% of all-time downloads
xlstm_wikipedia_110M_1M
PatrickHaller/xlstm_wikipedia_110M_1M
xlstm_wikipedia_110M_1M is a machine learning model from PatrickHaller. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
All-time downloads
250
Public
Parameters
111M
445 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors445 MB · 100%
At a glance
- License
- mit
- Model type
- xlstm
- Access
- Public
- Created
- Aug 23, 2024
- Updated
- Aug 23, 2024
- SHA
- ac2c1a8f
Model card
From the Hugging Face model README
An xLSTM Model
Trained with Helibrunna (fork)
To use this model the xLSTM package is required. We recommend to install it locally with conda:
git clone https://github.com/NX-AI/xlstm
cd xlstm
conda env create -n xlstm -f environment_pt220cu121.yaml
conda activate xlstm
Usage
from transformers import AutoModelForCasualLM, AutoTokenizer
model_name_or_path = "PatrickHaller/xlstm_wikipedia_110M_1M"
model = AutoModelForCasualLM.from_pretrained(model_name_or_path)
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
input_ids = tokenizer.encode("Hello, my dog is cute", return_tensors="pt")
output = model.generate(input_ids, max_length=100, temperature=0.7, do_sample=True)
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print(generated_text)
- Type
- xlstm
- License
- mit
- Languages
- en
- Created
- Aug 23, 2024
- Updated
- Aug 23, 2024