fix(tasks): Fixes a typo on generate_text.py script.

This commit is contained in:
Gustavo Rosa 2023-03-02 11:44:14 -03:00
Родитель 6408066473
Коммит 94aca143f8
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -10,12 +10,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Generates new tokens with a pre-trained model.")
parser.add_argument("pre_trained_model_path", type=str, help="Path to the pre-trained model file.")
parser.add_argument("pre_trained_model_path", type=str, help="Path to the pre-trained model path/file.")
parser.add_argument("prompt", type=str, help="Prompt to serve as the generation's context.")
parser.add_argument(
"-sbf", "--pre_trained_model_subfolder", type=str, default=None, help="Path to the pre-trained model file."
"-sf", "--pre_trained_model_subfolder", type=str, default=None, help="Subfolder to the pre-trained model path."
)
args = parser.parse_args()