PygmalionAI / gradio-ui

Prototype UI for chatting with the Pygmalion models.
GNU Affero General Public License v3.0
227 stars 67 forks source link

Use KoboldAI as an inference server, add notebook for Colab #6

Closed 0x000011b closed 1 year ago

0x000011b commented 1 year ago

What it says on the tin. We use Kobold as an inference server so we can fit a 6B model on a Colab GPU (16GB T4s as of the time I'm writing this), and offer a notebook to automatically start it up and launch the Gradio UI to talk to it.