Skip to content

Commit

Permalink
Fix the TOGETHER class
Browse files Browse the repository at this point in the history
Summary:
Together API changes the field `model_name` to `model` now:
```
pydantic.v1.error_wrappers.ValidationError: 2 validation errors for Together
model
  field required (type=value_error.missing)
model_name
  extra fields not permitted (type=value_error.extra)
```
Hence make the update.

Reviewed By: spencerwmeta

Differential Revision: D51925200

fbshipit-source-id: fb285d0ace46c71d572ce1e401a14bf0c55bcedc
  • Loading branch information
SimonWan authored and facebook-github-bot committed Dec 7, 2023
1 parent cff76b0 commit 0845e27
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion CybersecurityBenchmarks/benchmark/llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -177,7 +177,7 @@ class TOGETHER(LLM):
@override
def query(self, prompt: str) -> str:
llm = Together(
model_name=self.model,
model=self.model,
temperature=0.75,
top_p=1,
max_tokens=MAX_TOKENS,
Expand Down

0 comments on commit 0845e27

Please sign in to comment.