update cap

multimodal/open_flamingo/chat/conversation.py

@@ -274,7 +274,7 @@ def preprocess_image(sample, image_processor):
         image = torch.tensor(image["pixel_values"][0])
     return image
 
-
+@dataclasses.dataclass
 class ChatBOT:
     def __init__(self, model, vis_processor, tokenizer, vis_embed_size,model_name):
         self.model = model