lorocksUMD commited on
Commit
10ffe76
1 Parent(s): 7222efe

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +2 -2
script.py CHANGED
@@ -67,12 +67,12 @@ def load_images(image_files):
67
  return out
68
 
69
 
70
- model_name = get_model_name_from_path(model_path)
71
 
72
  args = type('Args', (), {
73
  "model_path": model_path,
74
  "model_base": None,
75
- "model_name": get_model_name_from_path(model_path),
76
  "query": prompt,
77
  "conv_mode": None,
78
  "image_file": image_file,
 
67
  return out
68
 
69
 
70
+ model_name = get_model_name_from_path('llava-v1.6-mistral-7b')
71
 
72
  args = type('Args', (), {
73
  "model_path": model_path,
74
  "model_base": None,
75
+ "model_name": model_name,
76
  "query": prompt,
77
  "conv_mode": None,
78
  "image_file": image_file,