Update custom_st.py
Browse files- custom_st.py +1 -1
    	
        custom_st.py
    CHANGED
    
    | @@ -69,7 +69,7 @@ class MultiModalTransformer(BaseTransformer): | |
| 69 | 
             
                                raise ValueError(f"Unknown image type {type(img)}")
         | 
| 70 | 
             
                        if "text" in item:
         | 
| 71 | 
             
                            if text:
         | 
| 72 | 
            -
                                text += "<|begin_of_text|> | 
| 73 | 
             
                            text += item["text"].lstrip()
         | 
| 74 |  | 
| 75 | 
             
                        return text, img
         | 
|  | |
| 69 | 
             
                                raise ValueError(f"Unknown image type {type(img)}")
         | 
| 70 | 
             
                        if "text" in item:
         | 
| 71 | 
             
                            if text:
         | 
| 72 | 
            +
                                text += "<|begin_of_text|>"
         | 
| 73 | 
             
                            text += item["text"].lstrip()
         | 
| 74 |  | 
| 75 | 
             
                        return text, img
         |