Skip to content

Commit b301cf0

Browse files
committed
fix: remove huggingface transformers due to platform requirement
1 parent 917e032 commit b301cf0

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

app/evaluation.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,14 @@ def setup_llm(config):
3030
max_tokens=config.max_new_token,
3131
openai_api_key=config.openai_api_key
3232
)
33-
elif config.mode == 'llama3': # NOTE: langchain_huggingface expected "linux/arm64"
34-
from langchain_huggingface import HuggingFaceEndpoint
35-
return HuggingFaceEndpoint(
36-
endpoint_url=config.endpoint_3_1_8B,
37-
max_new_tokens=config.max_new_token,
38-
temperature=config.temperature,
39-
huggingfacehub_api_token=config.huggingfacehub_api_token
40-
)
33+
# elif config.mode == 'llama3': # NOTE: langchain_huggingface expected "linux/arm64"
34+
# from langchain_huggingface import HuggingFaceEndpoint
35+
# return HuggingFaceEndpoint(
36+
# endpoint_url=config.endpoint_3_1_8B,
37+
# max_new_tokens=config.max_new_token,
38+
# temperature=config.temperature,
39+
# huggingfacehub_api_token=config.huggingfacehub_api_token
40+
# )
4141

4242

4343
def parse_last_boolean(response):

app/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@ pandas
1313
langchain
1414
langchain-openai
1515
python-dotenv
16-
langchain_huggingface
16+
# langchain_huggingface

0 commit comments

Comments
 (0)