We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent abd6d88 commit 76c38f1Copy full SHA for 76c38f1
tests/unit/steps/tasks/structured_outputs/test_outlines.py
@@ -59,6 +59,7 @@ class DummyUserTest(BaseModel):
59
"device_map": None,
60
"token": None,
61
"use_magpie_template": False,
62
+ "disable_cuda_device_placement": False,
63
"type_info": {
64
"module": "distilabel.llms.huggingface.transformers",
65
"name": "TransformersLLM",
@@ -85,6 +86,7 @@ class DummyUserTest(BaseModel):
85
86
87
88
89
90
91
92
0 commit comments