File tree 1 file changed +5
-2
lines changed
1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -3098,7 +3098,7 @@ class NanoLlavaChatHandler(Llava15ChatHandler):
3098
3098
"{% endif %}"
3099
3099
)
3100
3100
3101
- class Llama3VisionAlpha (Llava15ChatHandler ):
3101
+ class Llama3VisionAlphaChatHandler (Llava15ChatHandler ):
3102
3102
# question = "<image>" + q
3103
3103
3104
3104
# prompt = f"<|start_header_id|>user<|end_header_id|>\n\n{question}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
@@ -3159,6 +3159,10 @@ class Llama3VisionAlpha(Llava15ChatHandler):
3159
3159
"{% endif %}"
3160
3160
)
3161
3161
3162
+ # alias
3163
+ Llama3VisionAlpha = Llama3VisionAlphaChatHandler
3164
+
3165
+
3162
3166
@register_chat_completion_handler ("chatml-function-calling" )
3163
3167
def chatml_function_calling (
3164
3168
llama : llama .Llama ,
@@ -3193,7 +3197,6 @@ def chatml_function_calling(
3193
3197
llama_types .CreateChatCompletionResponse ,
3194
3198
Iterator [llama_types .CreateChatCompletionStreamResponse ],
3195
3199
]:
3196
- print (logprobs )
3197
3200
function_calling_template = (
3198
3201
"{% for message in messages %}"
3199
3202
"<|im_start|>{{ message.role }}\n "
You can’t perform that action at this time.
0 commit comments