Spaces:
Runtime error
Runtime error
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 |
|