Skip to content

Commit 5360f2c

Browse files
committed
update main py
1 parent 8e7e109 commit 5360f2c

File tree

1 file changed

+1
-1
lines changed
  • modules/openvino_code/server

1 file changed

+1
-1
lines changed

modules/openvino_code/server/main.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def main():
1414
args = get_parser().parse_args()
1515

1616
# temporary solution for cli args passing
17-
generator_dependency = get_generator_dependency(args.model, args.device, args.tokenizer_checkpoint, args.assistant)
17+
generator_dependency = get_generator_dependency(args.model, 'CPU', args.tokenizer_checkpoint, args.assistant)
1818
app.dependency_overrides[get_generator_dummy] = generator_dependency
1919

2020
uvicorn.run(app, host=args.host, port=args.port)

0 commit comments

Comments
 (0)