Spaces:
Runtime error
Runtime error
Commit
·
4c262cb
1
Parent(s):
80b1c78
Update chat.py
Browse files
chat.py
CHANGED
|
@@ -132,6 +132,8 @@ class Chat:
|
|
| 132 |
stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)
|
| 133 |
streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=15)
|
| 134 |
|
|
|
|
|
|
|
| 135 |
max_new_tokens = min(max_new_tokens, max_context_length - input_ids.shape[-1] - num_image_tokens - num_seg_tokens - num_depth_tokens)
|
| 136 |
|
| 137 |
if max_new_tokens < 1:
|
|
|
|
| 132 |
stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)
|
| 133 |
streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=15)
|
| 134 |
|
| 135 |
+
print(input_ids)
|
| 136 |
+
|
| 137 |
max_new_tokens = min(max_new_tokens, max_context_length - input_ids.shape[-1] - num_image_tokens - num_seg_tokens - num_depth_tokens)
|
| 138 |
|
| 139 |
if max_new_tokens < 1:
|