SucheRAG

Sleeping

alexkueck commited on Jul 5, 2024

Commit

fe4701a

verified ·

1 Parent(s): 2d8cb29

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -19,7 +19,7 @@ import operator
 from typing import Annotated, Sequence, TypedDict
 import pprint
-from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
 from sentence_transformers import SentenceTransformer, util
 from typing import List, Dict

 from typing import Annotated, Sequence, TypedDict
 import pprint
+from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForCausalLM
 from sentence_transformers import SentenceTransformer, util
 from typing import List, Dict