Commit3a5d3bb

authored

Fix device assignment for transformers (#607)

1 parent450959c commit3a5d3bbCopy full SHA for 3a5d3bb

File tree

-3

lines changed

-3

lines changed

Lines changed: 1 addition & 1 deletion

Some generated files are not rendered by default. Learn more aboutcustomizing how changed files appear on GitHub.

Lines changed: 1 addition & 1 deletion

Whitespace-only changes.

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -53,9 +53,10 @@ def default(self, obj):`
`53`	`53`	`deftransform(task,args,inputs,cache):`
`54`	`54`	`task=json.loads(task)`
`55`	`55`	`args=json.loads(args)`
`56`		`-args["device"]=assign_device(args.get("device"))`
`57`	`56`	`inputs=json.loads(inputs)`
`58`	`57`
	`58`	`+task["device"]=assign_device(task.get("device"))`
	`59`	`+`
`59`	`60`	`ifcache:`
`60`	`61`	`key=",".join([f"{key}:{val}"for (key,val)insorted(task.items())])`
`61`	`62`	`ifkeynotin__cache_transform_pipeline_by_task:`

Comments

(0)