Skip to content

Commit

Permalink
Merge branch 'main' into rename-model-param-embedders
Browse files Browse the repository at this point in the history
  • Loading branch information
ZanSara authored Jan 12, 2024
2 parents ba531f8 + 24c71bd commit f073830
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -184,7 +184,7 @@ def test_to_dict_with_parameters(self):

def test_to_dict_with_quantization_config(self):
component = HuggingFaceLocalGenerator(
model_name_or_path="gpt2",
model="gpt2",
task="text-generation",
device="cuda:0",
token="test-token",
Expand Down

0 comments on commit f073830

Please sign in to comment.