coztomate commited on
Commit
1cb1c8e
1 Parent(s): e709466

update config

Browse files
Files changed (1) hide show
  1. config_llm.py +4 -4
config_llm.py CHANGED
@@ -1,13 +1,13 @@
1
  # Configuration and Constants
2
 
3
- # Defaults for Mistral/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
- # Mistral Model Configuration
9
- model_id_mistral = "mistralai/Mixtral-8x7B-Instruct-v0.1"
10
- API_URL = f"https://api-inference.huggingface.co/models/{model_id_mistral}"
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