Skip to content

Commit f77d2ba

Browse files
committed
update toml file and main.py
1 parent fa94818 commit f77d2ba

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-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, 'CPU', args.tokenizer_checkpoint, args.assistant)
17+
generator_dependency = get_generator_dependency(args.model, args.device, 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)

modules/openvino_code/server/pyproject.toml

+2
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@ build-backend = "setuptools.build_meta"
2828
[tool.black]
2929
line-length = 119
3030
target-version = ['py38', 'py39', 'py310', 'py311']
31+
unstable = true
32+
preview = true
3133

3234
[tool.ruff]
3335
lint.ignore = ["C901", "E501", "E741", "W605", "F401", "W292"]

0 commit comments

Comments
 (0)