Skip to content

Commit

Permalink
Update colab
Browse files Browse the repository at this point in the history
  • Loading branch information
Cohee1207 committed Nov 20, 2023
1 parent 41db146 commit af44a63
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions colab/GPU.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,9 @@
"#@markdown ***\n",
"#@markdown Enables Whisper speech recognition module\n",
"extras_enable_whisper_stt = True #@param {type:\"boolean\"}\n",
"whisper_model = \"base.en\" #@param [ \"tiny.en\", \"base.en\", \"small.en\", \"medium.en\", \"tiny\", \"base\", \"small\", \"medium\", \"large\" ]\n",
"#@markdown There are five model sizes, four with English-only versions, offering speed and accuracy tradeoffs.\n",
"#@markdown The .en models for English-only applications tend to perform better, especially for the tiny.en and base.en models.\n",
"#@markdown ***\n",
"#@markdown Enables SD picture generation\n",
"extras_enable_sd = True #@param {type:\"boolean\"}\n",
Expand All @@ -80,13 +83,9 @@
"#@markdown * prompthero/openjourney - midjourney style model\n",
"#@markdown * ckpt/sd15 - base SD 1.5\n",
"#@markdown * stabilityai/stable-diffusion-2-1-base - base SD 2.1\n",
"whisper_model = \"base.en\" #@param [ \"tiny.en\", \"base.en\", \"small.en\", \"medium.en\", \"tiny\", \"base\", \"small\", \"medium\", \"large\" ]\n",
"#@markdown There are five model sizes, four with English-only versions, offering speed and accuracy tradeoffs.\n",
"#@markdown The .en models for English-only applications tend to perform better, especially for the tiny.en and base.en models.\n",
"#@markdown ***\n",
"#@markdown Enables TalkingHead module\n",
"extras_enable_talkinghead = False #@param {type:\"boolean\"}\n",
"#@markdown ***\n",
"#@markdown Enables ChromaDB module\n",
"extras_enable_chromadb = True #@param {type:\"boolean\"}\n",
"\n",
Expand Down

0 comments on commit af44a63

Please sign in to comment.