Update constants.py
Browse files- constants.py +1 -1
constants.py
CHANGED
@@ -3,7 +3,7 @@ DEFAULT_CUDA_DEVICE = "cuda:0"
|
|
3 |
# Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
|
4 |
DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"
|
5 |
# Also try: 'joeddav/distilbert-base-uncased-go-emotions-student'
|
6 |
-
DEFAULT_CLASSIFICATION_MODEL = "
|
7 |
# Also try: 'Salesforce/blip-image-captioning-base'
|
8 |
DEFAULT_CAPTIONING_MODEL = "Salesforce/blip-image-captioning-large"
|
9 |
DEFAULT_SD_MODEL = "ckpt/anything-v4.5-vae-swapped"
|
|
|
3 |
# Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
|
4 |
DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"
|
5 |
# Also try: 'joeddav/distilbert-base-uncased-go-emotions-student'
|
6 |
+
DEFAULT_CLASSIFICATION_MODEL = "joeddav/distilbert-base-uncased-go-emotions-student"
|
7 |
# Also try: 'Salesforce/blip-image-captioning-base'
|
8 |
DEFAULT_CAPTIONING_MODEL = "Salesforce/blip-image-captioning-large"
|
9 |
DEFAULT_SD_MODEL = "ckpt/anything-v4.5-vae-swapped"
|