Italia-9B / SCRIPT_README.md
leafspark's picture
add model
56811f1 verified
```python
# Modello Italia inference script and model
# Copyright 2024 iGenius
#
# Licensed under the MIT License (see LICENSE-MIT).
# This code also contains code from the original project licensed under the Apache License 2.0 (see LICENSE-APACHE).
# This script contains modifications of the original code from Lightning AI.
```
### Instructions
1. First, move the model and the tokenizer from `/modello_italia_9b` to the current directory, or ensure that the path is correctly specified.
2. Install dependencies by running the following command in the terminal:
```terminal
pip install -r requirements.txt
```
3. To run the generation, use the following command:
```terminal
python generate.py --checkpoint_dir <model_path> --max_new_tokens 500 --temperature 0.2 --prompt "Ciao, chi sei?"
```