Instructions to use ModelSpace/GemmaX2-28-2B-v0.1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ModelSpace/GemmaX2-28-2B-v0.1 with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "translation" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # pip install "transformers<5.0.0" from transformers import pipeline pipe = pipeline("translation", model="ModelSpace/GemmaX2-28-2B-v0.1")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("ModelSpace/GemmaX2-28-2B-v0.1") model = AutoModelForCausalLM.from_pretrained("ModelSpace/GemmaX2-28-2B-v0.1", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update generation_config.json
Browse files- generation_config.json +2 -1
generation_config.json
CHANGED
|
@@ -4,5 +4,6 @@
|
|
| 4 |
"cache_implementation": "hybrid",
|
| 5 |
"eos_token_id": 1,
|
| 6 |
"pad_token_id": 0,
|
| 7 |
-
"transformers_version": "4.45.0"
|
|
|
|
| 8 |
}
|
|
|
|
| 4 |
"cache_implementation": "hybrid",
|
| 5 |
"eos_token_id": 1,
|
| 6 |
"pad_token_id": 0,
|
| 7 |
+
"transformers_version": "4.45.0",
|
| 8 |
+
"max_new_tokens": 512
|
| 9 |
}
|