gojiteji commited on
Commit
da31fca
1 Parent(s): ab4f992

support llama2

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -12,7 +12,10 @@ def greet(modelpath1="",modelpath2="",modelpath3="",modelpath4="",modelpath5="")
12
  return "Please enter at least 2 model names."
13
  urls = []
14
  for name in names:
15
- urls.append("https://huggingface.co/" + name + "/raw/main/config.json")
 
 
 
16
 
17
  configs = []
18
  index_to_ignore =[]
 
12
  return "Please enter at least 2 model names."
13
  urls = []
14
  for name in names:
15
+ if name in ["meta-llama/Llama-2-7b-hf","meta-llama/Llama-2-13b-hf","meta-llama/Llama-2-70b-hf","meta-llama/Llama-2-7b-chat-hf","meta-llama/Llama-2-13b-chat-hf","meta-llama/Llama-2-70b-chat-hf"]:
16
+ urls.append("https://huggingface.co/spaces/gojiteji/LLM-Comparer/blob/main/" + name + "/config.json")
17
+ else:
18
+ urls.append("https://huggingface.co/" + name + "/raw/main/config.json")
19
 
20
  configs = []
21
  index_to_ignore =[]