Downloads · 30 days
21
15% of all-time downloads
xlstm_wikipedia_110M_100M
PatrickHaller/xlstm_wikipedia_110M_100M
xlstm_wikipedia_110M_100M 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
140
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 26, 2024
- Updated
- Aug 26, 2024
- SHA
- cca7565e
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_dummy"
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 26, 2024
- Updated
- Aug 26, 2024