Spaces:
Runtime error
Runtime error
LanguageBind
commited on
Commit
•
f6e0af0
1
Parent(s):
2ce07bc
Update llava/serve/gradio_utils.py
Browse files
llava/serve/gradio_utils.py
CHANGED
@@ -105,7 +105,7 @@ class Chat:
|
|
105 |
else:
|
106 |
input_ids = tokenizer_X_token(prompt, tokenizer, X_TOKEN_INDEX['IMAGE'], return_tensors='pt').unsqueeze(0).to(self.device)
|
107 |
|
108 |
-
temperature = 0.
|
109 |
|
110 |
max_new_tokens = 1024
|
111 |
|
|
|
105 |
else:
|
106 |
input_ids = tokenizer_X_token(prompt, tokenizer, X_TOKEN_INDEX['IMAGE'], return_tensors='pt').unsqueeze(0).to(self.device)
|
107 |
|
108 |
+
temperature = 0.1
|
109 |
|
110 |
max_new_tokens = 1024
|
111 |
|