Shago commited on
Commit
ae8c668
·
verified ·
1 Parent(s): 0de5606

Update llm.py

Browse files

langchain_huggingface

Files changed (1) hide show
  1. llm.py +1 -1
llm.py CHANGED
@@ -2,7 +2,7 @@ from transformers import pipeline
2
  import torch
3
  from langchain.prompts import ChatPromptTemplate
4
  from langchain.schema.output_parser import StrOutputParser
5
- from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
6
 
7
  # Initialize HF pipeline for text generation
8
  text_generator = pipeline(
 
2
  import torch
3
  from langchain.prompts import ChatPromptTemplate
4
  from langchain.schema.output_parser import StrOutputParser
5
+ from langchain_huggingface import HuggingFacePipeline
6
 
7
  # Initialize HF pipeline for text generation
8
  text_generator = pipeline(