removed device to work on cpu

philschmid · philschmid · commit 41a1892784bf · 2021-07-21T17:33:50.000+02:00
diff --git a/tests/unit/test_transformers_utils.py b/tests/unit/test_transformers_utils.py
@@ -133,7 +133,6 @@ def test_wrap_conversation_pipeline():
         model="microsoft/DialoGPT-small",
         tokenizer="microsoft/DialoGPT-small",
         framework="pt",
-        device=0,
     )
     conv_pipe = wrap_conversation_pipeline(init_pipeline)
     data = {

Original file line number	Diff line number	Diff line change
`@@ -133,7 +133,6 @@ def test_wrap_conversation_pipeline():`
`133`	`133`	`model="microsoft/DialoGPT-small",`
`134`	`134`	`tokenizer="microsoft/DialoGPT-small",`
`135`	`135`	`framework="pt",`
`136`		`- device=0,`
`137`	`136`	`)`
`138`	`137`	`conv_pipe = wrap_conversation_pipeline(init_pipeline)`
`139`	`138`	`data = {`