mirror of
https://github.com/bigscience-workshop/petals
synced 2024-10-31 09:20:41 +00:00
Merge branch 'main' into bnb-0-41-1
This commit is contained in:
commit
16e6ce95e8
@ -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",
|
||||||
|
Loading…
Reference in New Issue
Block a user