File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -24,7 +24,7 @@ def processor_of_block(block: GraniteioModelBlock):
24
24
assert isinstance (model , str ), f"The model should be a string: { model } "
25
25
assert isinstance (
26
26
backend , (dict , str )
27
- ), f"The backend should be a string or a dictionnary : { backend } "
27
+ ), f"The backend should be a string or a dictionary : { backend } "
28
28
match backend :
29
29
case {"transformers" : device }:
30
30
assert isinstance (backend , dict )
@@ -83,7 +83,7 @@ async def async_generate_text(
83
83
assert parameters is None or isinstance (parameters , dict )
84
84
io_processor = GraniteioModel .processor_of_block (block )
85
85
inputs = GraniteioModel .build_message (messages , parameters )
86
- result = io_processor .create_chat_completion (inputs ) # pyright: ignore
86
+ result = await io_processor .acreate_chat_completion (inputs ) # pyright: ignore
87
87
try : # TODO: update when new version of granite-io is released
88
88
message = result .next_message .model_dump ()
89
89
except AttributeError :
You can’t perform that action at this time.
0 commit comments