Chamin09 commited on
Commit
d776322
·
verified ·
1 Parent(s): 2202e15

Update models/llm_setup.py

Browse files
Files changed (1) hide show
  1. models/llm_setup.py +3 -3
models/llm_setup.py CHANGED
@@ -1,7 +1,7 @@
1
  # Updated import path
2
- #from llama_index.llms import HuggingFaceInferenceAPI
3
- from llama_index.llms.huggingface import HuggingFaceLLM
4
- from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
5
  import torch
6
  # If that doesn't work, try:
7
  # from llama_index.llms.huggingface import HuggingFaceLLM
 
1
  # Updated import path
2
+ from llama_index.llms import HuggingFaceInferenceAPI
3
+ #from llama_index.llms.huggingface import HuggingFaceLLM
4
+ #from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
5
  import torch
6
  # If that doesn't work, try:
7
  # from llama_index.llms.huggingface import HuggingFaceLLM