pabloce commited on
Commit
59c24a4
·
verified ·
1 Parent(s): b742530

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -19,8 +19,8 @@ hf_hub_download(
19
  local_dir="./models"
20
  )
21
  hf_hub_download(
22
- repo_id="bartowski/Llama-3-8B-Instruct-262k-GGUF",
23
- filename="Llama-3-8B-Instruct-262k-Q6_K.gguf",
24
  local_dir="./models"
25
  )
26
 
@@ -91,7 +91,7 @@ def get_messages_formatter_type(model_name):
91
  return MessagesFormatterType.LLAMA_3
92
  elif "Mistral" in model_name:
93
  return MessagesFormatterType.MISTRAL
94
- elif "mixtral" in model_name:
95
  return MessagesFormatterType.MISTRAL
96
  elif "Phi" in model_name:
97
  return MessagesFormatterType.PHI_3
@@ -215,7 +215,7 @@ demo = gr.ChatInterface(
215
  ),
216
  gr.Dropdown([
217
  'Mistral-7B-Instruct-v0.3-f32.gguf',
218
- 'Llama-3-8B-Instruct-262k-Q6_K.gguf'
219
  ],
220
  value="Mistral-7B-Instruct-v0.3-f32.gguf",
221
  label="Model"
 
19
  local_dir="./models"
20
  )
21
  hf_hub_download(
22
+ repo_id="bartowski/Einstein-v6-7B-GGUF",
23
+ filename="Einstein-v6-7B-Q6_K.gguf",
24
  local_dir="./models"
25
  )
26
 
 
91
  return MessagesFormatterType.LLAMA_3
92
  elif "Mistral" in model_name:
93
  return MessagesFormatterType.MISTRAL
94
+ elif "Einstein-v6-7B" in model_name:
95
  return MessagesFormatterType.MISTRAL
96
  elif "Phi" in model_name:
97
  return MessagesFormatterType.PHI_3
 
215
  ),
216
  gr.Dropdown([
217
  'Mistral-7B-Instruct-v0.3-f32.gguf',
218
+ 'Einstein-v6-7B-Q6_K.gguf'
219
  ],
220
  value="Mistral-7B-Instruct-v0.3-f32.gguf",
221
  label="Model"