Skip to content

Commit 76c38f1

Browse files
committed
Fix unit test
1 parent abd6d88 commit 76c38f1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/steps/tasks/structured_outputs/test_outlines.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ class DummyUserTest(BaseModel):
5959
"device_map": None,
6060
"token": None,
6161
"use_magpie_template": False,
62+
"disable_cuda_device_placement": False,
6263
"type_info": {
6364
"module": "distilabel.llms.huggingface.transformers",
6465
"name": "TransformersLLM",
@@ -85,6 +86,7 @@ class DummyUserTest(BaseModel):
8586
"device_map": None,
8687
"token": None,
8788
"use_magpie_template": False,
89+
"disable_cuda_device_placement": False,
8890
"type_info": {
8991
"module": "distilabel.llms.huggingface.transformers",
9092
"name": "TransformersLLM",

0 commit comments

Comments
 (0)