Lahiru Menikdiwela
commited on
Commit
·
70f6a45
1
Parent(s):
72b02de
fix automodel for llama -pipepline
Browse files
model.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import os
|
| 2 |
-
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, AutoModelForCausalLM
|
| 3 |
from transformers import LEDForConditionalGeneration, LEDTokenizer
|
| 4 |
from langchain_openai import OpenAI
|
| 5 |
# from huggingface_hub import login
|
|
|
|
| 1 |
import os
|
| 2 |
+
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline, AutoModelForCausalLM
|
| 3 |
from transformers import LEDForConditionalGeneration, LEDTokenizer
|
| 4 |
from langchain_openai import OpenAI
|
| 5 |
# from huggingface_hub import login
|