Downloads · 30 days
16
6% of all-time downloads
torchtune_test
smohammadi/torchtune_test
torchtune_test is a machine learning model from smohammadi. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is a finetuned version of google/gemma-2b
All-time downloads
272
Public
Repo size
295 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin148 MB · 50%
At a glance
- Model type
- gemma
- Access
- Public
- Created
- Oct 6, 2024
- Updated
- Oct 6, 2024
- SHA
- ecc19722
Base models
Datasets
Model card
From the Hugging Face model README
My Torchtune Model
This model is a finetuned version of google/gemma-2b
Model description
More information needed
Training and evaluation results
More information needed
Training procedure
This model was trained using the torchtune library using the following command:
/Users/salmanmohammadi/projects/torchtune/recipes/lora_finetune_single_device.py --config /Users/salmanmohammadi/projects/torchtune/recipes/configs/gemma/2B_lora_single_device.yaml \
device=mps \
epochs=1 \
max_steps_per_epoch=10
Framework versions
- torchtune 0.0.0
- torchao 0.5.0
- datasets 2.20.0
- sentencepiece 0.2.0
- Type
- gemma
- Languages
- en
- Created
- Oct 6, 2024
- Updated
- Oct 6, 2024