chendl commited on
Commit
844a1e9
·
1 Parent(s): e82e643

update chat

Browse files
multimodal/open_flamingo/chat/conversation.py CHANGED
@@ -4,6 +4,7 @@ from PIL import Image
4
 
5
  import torch
6
  import numpy as np
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaTokenizer
8
  from transformers import StoppingCriteria, StoppingCriteriaList
9
 
 
4
 
5
  import torch
6
  import numpy as np
7
+ import transformers
8
  from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaTokenizer
9
  from transformers import StoppingCriteria, StoppingCriteriaList
10