update config
Browse files- config_llm.py +4 -4
config_llm.py
CHANGED
@@ -1,13 +1,13 @@
|
|
1 |
# Configuration and Constants
|
2 |
|
3 |
-
# Defaults for
|
4 |
DEFAULT_SYSTEM_PROMPT = "You will be given a very short description of a contemporary artwork. Please create a complex exhibition press text based on the given artwork description using international art english dealing with post-colonialism, military industrial complex, anthropocene, identity politics and queerness through the language of Rancière, Fontane, Paglen, Deleuze, Steyerl, Spivak, Preciado, Žižek, Foucault and Harraway. Avoid excessive namedropping. Just output press text without explaining your actions."
|
5 |
EOS_STRING = "</s>"
|
6 |
EOT_STRING = "<EOT>"
|
7 |
|
8 |
-
#
|
9 |
-
|
10 |
-
API_URL = f"https://api-inference.huggingface.co/models/{
|
11 |
|
12 |
#model parameters
|
13 |
MAX_NEW_TOKENS = 2048
|
|
|
1 |
# Configuration and Constants
|
2 |
|
3 |
+
# Defaults for Mixtral/OpenAI
|
4 |
DEFAULT_SYSTEM_PROMPT = "You will be given a very short description of a contemporary artwork. Please create a complex exhibition press text based on the given artwork description using international art english dealing with post-colonialism, military industrial complex, anthropocene, identity politics and queerness through the language of Rancière, Fontane, Paglen, Deleuze, Steyerl, Spivak, Preciado, Žižek, Foucault and Harraway. Avoid excessive namedropping. Just output press text without explaining your actions."
|
5 |
EOS_STRING = "</s>"
|
6 |
EOT_STRING = "<EOT>"
|
7 |
|
8 |
+
# Mixtral Model Configuration
|
9 |
+
model_id_mixtral = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
10 |
+
API_URL = f"https://api-inference.huggingface.co/models/{model_id_mixtral}"
|
11 |
|
12 |
#model parameters
|
13 |
MAX_NEW_TOKENS = 2048
|