Downloads · 30 days
26
17% of all-time downloads
mamba_0_875_sft
JunxiongWang/mamba_0_875_sft
mamba_0_875_sft is a text generation model from JunxiongWang. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
All-time downloads
157
Public
Repo size
62.2 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin31.1 GB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- mit
- Model type
- mistral
- Access
- Public
- Created
- Jul 18, 2024
- Updated
- Jul 19, 2024
- SHA
- 453c20d0
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
Please check here for details.
<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->mamba_0_875_sft
This model is a fine-tuned version of JunxiongWang/mamba_0_75_sft on the JunxiongWang/sftdataset dataset. It achieves the following results on the evaluation set:
- Loss: 0.5037
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 8
- eval_batch_size: 4
- seed: 42
- distributed_type: multi-GPU
- num_devices: 8
- total_train_batch_size: 64
- total_eval_batch_size: 32
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: cosine
- lr_scheduler_warmup_ratio: 0.01
- num_epochs: 1
Training results
| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.6117 | 1.0 | 55064 | 0.5037 |
Framework versions
- Transformers 4.41.2
- Pytorch 2.1.0+cu118
- Datasets 2.20.0
- Tokenizers 0.19.1
- Task
- Text Generation
- Library
- transformers
- Type
- mistral
- License
- mit
- Created
- Jul 18, 2024
- Updated
- Jul 19, 2024