brendon-ai commited on
Commit
0ec7bbc
·
verified ·
1 Parent(s): afb2b95

Update src/RAGSample.py

Browse files
Files changed (1) hide show
  1. src/RAGSample.py +3 -1
src/RAGSample.py CHANGED
@@ -387,6 +387,8 @@ Answer:
387
  )
388
 
389
  # Initialize a local Hugging Face model
 
 
390
  hf_pipeline = pipeline(
391
  "text-generation",
392
  model="microsoft/BioGPT",
@@ -401,7 +403,7 @@ Answer:
401
  do_sample=True,
402
  pad_token_id=1,
403
  eos_token_id=2,
404
- )
405
 
406
  # Wrap it in LangChain
407
  llm = HuggingFacePipeline(pipeline=hf_pipeline)
 
387
  )
388
 
389
  # Initialize a local Hugging Face model
390
+ hf_pipeline = pipeline(
391
+ # Initialize a local Hugging Face model
392
  hf_pipeline = pipeline(
393
  "text-generation",
394
  model="microsoft/BioGPT",
 
403
  do_sample=True,
404
  pad_token_id=1,
405
  eos_token_id=2,
406
+ )
407
 
408
  # Wrap it in LangChain
409
  llm = HuggingFacePipeline(pipeline=hf_pipeline)