diff --git a/colab/TPU.ipynb b/colab/TPU.ipynb index 4a6efcd9..3976f6dd 100644 --- a/colab/TPU.ipynb +++ b/colab/TPU.ipynb @@ -66,7 +66,7 @@ "#@title <-- Select your model below and then click this to start KoboldAI\n", "#@markdown You can find a description of the models below along with instructions on how to start KoboldAI.\n", "\n", - "Model = \"Nerys 13B V2\" #@param [\"Nerys 13B V2\", \"Erebus 13B\", \"Janeway 13B\", \"Shinen 13B\", \"Skein 20B\", \"Erebus 20B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Shinen 6B\", \"Pygmalion 6B\", \"Lit V2 6B\", \"Lit 6B\", \"NeoX 20B\", \"OPT 13B\", \"Fairseq Dense 13B\", \"GPT-J-6B\"] {allow-input: true}\n", + "Model = \"Nerys 13B V2\" #@param [\"Nerys 13B V2\", \"Nerybus 13B\", \"Erebus 13B\", \"Janeway 13B\", \"Shinen 13B\", \"Skein 20B\", \"Erebus 20B\", \"Skein 6B\", \"Janeway 6B\", \"Adventure 6B\", \"Shinen 6B\", \"Pygmalion 6B\", \"Lit V2 6B\", \"Lit 6B\", \"NeoX 20B\", \"OPT 13B\", \"Fairseq Dense 13B\", \"GPT-J-6B\"] {allow-input: true}\n", "Version = \"Official\" #@param [\"Official\", \"United\"] {allow-input: true}\n", "Provider = \"Cloudflare\" #@param [\"Localtunnel\", \"Cloudflare\"]\n", "use_google_drive = True #@param {type:\"boolean\"}\n", @@ -97,6 +97,10 @@ " Model = \"KoboldAI/OPT-13B-Nerys-v2\"\n", " path = \"\"\n", " download = \"\"\n", + "elif Model == \"Nerybus 13B\":\n", + " Model = \"KoboldAI/OPT-13B-Nerybus-Mix\"\n", + " path = \"\"\n", + " download = \"\"\n", "elif Model == \"Erebus 13B\":\n", " Model = \"KoboldAI/OPT-13B-Erebus\"\n", " path = \"\"\n", diff --git a/environments/huggingface.yml b/environments/huggingface.yml index d1aca206..222bb6ad 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -30,7 +30,7 @@ dependencies: - flask-cors - lupa==1.10 - transformers==4.25.1 - - huggingface_hub>=0.10.1 + - huggingface_hub==0.12.1 - safetensors - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra diff --git a/environments/rocm.yml b/environments/rocm.yml index ad6ea39a..13eae816 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -29,7 +29,7 @@ dependencies: - flask-cors - lupa==1.10 - transformers==4.25.1 - - huggingface_hub>=0.10.1 + - huggingface_hub==0.12.1 - safetensors - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra diff --git a/requirements.txt b/requirements.txt index cdde17c1..e7f79dbd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ transformers==4.25.1 -huggingface_hub>=0.10.1 +huggingface_hub==0.12.1 Flask Flask-SocketIO requests diff --git a/requirements_mtj.txt b/requirements_mtj.txt index e5488be8..e0f89ece 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -6,7 +6,7 @@ dm-haiku == 0.0.5 jax == 0.2.21 jaxlib >= 0.1.69, <= 0.3.7 transformers == 4.25.1 -huggingface_hub >= 0.10.1 +huggingface_hub==0.12.1 progressbar2 git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck flask