Merge branch 'main' into bnb-0-41-1

This commit is contained in:
Alexander Borzunov 2023-08-07 02:23:12 +04:00 committed by GitHub
commit 16e6ce95e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -92,9 +92,6 @@
}, },
"outputs": [], "outputs": [],
"source": [ "source": [
"# Choose a model you'd like to prompt-tune. We recommend starting with\n",
"# a smaller model (bigscience/bloom-7b1-petals) for faster prototyping.\n",
"# The code below uses LLaMA-65B.\n",
"MODEL_NAME = \"enoch/llama-65b-hf\"\n", "MODEL_NAME = \"enoch/llama-65b-hf\"\n",
"\n", "\n",
"# Choose a prompt-tuning mode ('ptune' or 'deep_ptune').\n", "# Choose a prompt-tuning mode ('ptune' or 'deep_ptune').\n",