chenzizhao commited on
Commit
3e323c0
1 Parent(s): 87b7a45

set padding=True

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -48,7 +48,7 @@ def get_model_response( # predict
48
  new_chats = chats + [chat]
49
  currently_selected = previous_selected[-1] if len(previous_selected) > 0 else []
50
  model_input: Dict[str, Any] = adapter.compose(
51
- image_paths, new_chats, previous_selected, True, False)
52
  model_input = nested_to_device(model_input)
53
 
54
  with torch.inference_mode(), torch.autocast(device_type=device().type,
 
48
  new_chats = chats + [chat]
49
  currently_selected = previous_selected[-1] if len(previous_selected) > 0 else []
50
  model_input: Dict[str, Any] = adapter.compose(
51
+ image_paths, new_chats, previous_selected, True, True)
52
  model_input = nested_to_device(model_input)
53
 
54
  with torch.inference_mode(), torch.autocast(device_type=device().type,
requirements.txt CHANGED
@@ -7,3 +7,4 @@ peft==0.10.0
7
  nltk==3.8.1
8
  gradio==4.44.1
9
  spaces==0.30.4
 
 
7
  nltk==3.8.1
8
  gradio==4.44.1
9
  spaces==0.30.4
10
+ numpy==1.26.4