add gguf path to generate

pytorch · Apr 17, 2024 · 9563191 · 9563191
1 parent da53078
commit 9563191
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/et.yml b/.github/workflows/et.yml
@@ -134,7 +134,7 @@ jobs:
           export GGUF_TOKENIZER_PATH=gguf_files/tokenizer.model
 
           python torchchat.py export --gguf-path ${GGUF_PATH} --output-pte-path ${PWD}/${MODEL_NAME}.pte
-          python torchchat.py generate --pte-path ${PWD}/${MODEL_NAME}.pte --tokenizer-path ${GGUF_TOKENIZER_PATH} --max-new-tokens 10 --temperature 0 > ${PWD}/output_et
+          python torchchat.py generate --gguf-path ${GGUF_PATH} --pte-path ${PWD}/${MODEL_NAME}.pte --tokenizer-path ${GGUF_TOKENIZER_PATH} --temperature 0 --max-new-tokens 20 > ${PWD}/output_et
           cat ${PWD}/output_et
 
           echo "Tests complete."