Spaces:
Runtime error
Runtime error
Commit
·
1cf9223
1
Parent(s):
570b0f3
Update constants.py
Browse files- constants.py +2 -2
constants.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
# Constants
|
2 |
DEFAULT_CUDA_DEVICE = "cuda:0"
|
3 |
# Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
|
4 |
-
DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-
|
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"
|
|
|
1 |
# Constants
|
2 |
DEFAULT_CUDA_DEVICE = "cuda:0"
|
3 |
# Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
|
4 |
+
DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ElectrifAi_v14"
|
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"
|