lorocksUMD commited on
Commit
3eb49db
1 Parent(s): 431dab7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -12
app.py CHANGED
@@ -97,18 +97,9 @@ else:
97
  else:
98
  qs = DEFAULT_IMAGE_TOKEN + "\n" + qs
99
 
100
- if "llama-2" in model_name.lower():
101
- conv_mode = "llava_llama_2"
102
- elif "mistral" in model_name.lower():
103
- conv_mode = "mistral_instruct"
104
- elif "v1.6-34b" in model_name.lower():
105
- conv_mode = "chatml_direct"
106
- elif "v1" in model_name.lower():
107
- conv_mode = "llava_v1"
108
- elif "mpt" in model_name.lower():
109
- conv_mode = "mpt"
110
- else:
111
- conv_mode = "llava_v0"
112
 
113
  if args.conv_mode is not None and conv_mode != args.conv_mode:
114
  print(
 
97
  else:
98
  qs = DEFAULT_IMAGE_TOKEN + "\n" + qs
99
 
100
+
101
+ conv_mode = "mistral_instruct"
102
+
 
 
 
 
 
 
 
 
 
103
 
104
  if args.conv_mode is not None and conv_mode != args.conv_mode:
105
  print(