Add chat template to processor_config.json (#36)
Browse files- Add chat template to processor_config.json (f38a3b95248373646993deed003c692b649246c1)
Co-authored-by: Matthew Carrigan <[email protected]>
- processor_config.json +1 -0
 
    	
        processor_config.json
    CHANGED
    
    | 
         @@ -1,4 +1,5 @@ 
     | 
|
| 1 | 
         
             
            {
         
     | 
| 
         | 
|
| 2 | 
         
             
              "image_seq_len": 64,
         
     | 
| 3 | 
         
             
              "processor_class": "Idefics2Processor"
         
     | 
| 4 | 
         
             
            }
         
     | 
| 
         | 
|
| 1 | 
         
             
            {
         
     | 
| 2 | 
         
            +
              "chat_template": "{% for message in messages %}{{message['role'].capitalize()}}{% if message['content'][0]['type'] == 'image' %}{{':'}}{% else %}{{': '}}{% endif %}{% for line in message['content'] %}{% if line['type'] == 'text' %}{{line['text']}}{% elif line['type'] == 'image' %}{{ '<image>' }}{% endif %}{% endfor %}<end_of_utterance>\n{% endfor %}{% if add_generation_prompt %}{{ 'Assistant:' }}{% endif %}",
         
     | 
| 3 | 
         
             
              "image_seq_len": 64,
         
     | 
| 4 | 
         
             
              "processor_class": "Idefics2Processor"
         
     | 
| 5 | 
         
             
            }
         
     |