Spaces:
Running
on
Zero
Running
on
Zero
chenzizhao
commited on
Commit
•
3e323c0
1
Parent(s):
87b7a45
set padding=True
Browse files- app.py +1 -1
- 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,
|
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
|