From cb3b6f48882ef98debaa8045c3bc71df436739bc Mon Sep 17 00:00:00 2001 From: SillyLossy Date: Thu, 9 Mar 2023 23:43:09 +0200 Subject: [PATCH] Update go-emotion option in colab to distilbert --- colab/GPU.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb index 53787d8..4ab566d 100644 --- a/colab/GPU.ipynb +++ b/colab/GPU.ipynb @@ -75,9 +75,9 @@ "#@markdown * Salesforce/blip-image-captioning-base - slightly faster but less accurate\n", "#@markdown * microsoft/git-large-r-textcaps - works best for pictures with text (memes), but slightly worse for cartoon/anime drawings\n", "extras_enable_emotions = True #@param {type:\"boolean\"}\n", - "Emotions_Model = \"bhadresh-savani/distilbert-base-uncased-emotion\" #@param [\"bhadresh-savani/distilbert-base-uncased-emotion\", \"bhadresh-savani/bert-base-go-emotion\"]\n", + "Emotions_Model = \"bhadresh-savani/distilbert-base-uncased-emotion\" #@param [\"bhadresh-savani/distilbert-base-uncased-emotion\", \"joeddav/distilbert-base-uncased-go-emotions-student\"]\n", "#@markdown * bhadresh-savani/distilbert-base-uncased-emotion = 6 supported emotions
\n", - "#@markdown * bhadresh-savani/bert-base-go-emotion = 28 supported emotions\n", + "#@markdown * joeddav/distilbert-base-uncased-go-emotions-student = 28 supported emotions\n", "extras_enable_memory = True #@param {type:\"boolean\"}\n", "Memory_Model = \"Qiliang/bart-large-cnn-samsum-ChatGPT_v3\" #@param [ \"Qiliang/bart-large-cnn-samsum-ChatGPT_v3\", \"Qiliang/bart-large-cnn-samsum-ElectrifAi_v10\", \"distilbart-xsum-12-3\" ]\n", "#@markdown * Qiliang/bart-large-cnn-samsum-ChatGPT_v3 - summarization model optimized for chats\n",