@@ -419,7 +419,7 @@ def run(
419
419
model_name : str ,
420
420
* ,
421
421
account_id : str ,
422
- image : Iterable [float ],
422
+ image : Union [ Iterable [float ], str ],
423
423
max_tokens : int | NotGiven = NOT_GIVEN ,
424
424
messages : Iterable [ai_run_params .ImageToTextMessage ] | NotGiven = NOT_GIVEN ,
425
425
prompt : str | NotGiven = NOT_GIVEN ,
@@ -475,7 +475,7 @@ def run(
475
475
text : str | Union [str , List [str ]] | NotGiven = NOT_GIVEN ,
476
476
prompt : str | NotGiven = NOT_GIVEN ,
477
477
guidance : float | NotGiven = NOT_GIVEN ,
478
- image : Iterable [float ] | NotGiven = NOT_GIVEN ,
478
+ image : Iterable [float ] | Union [ Iterable [ float ], str ] | NotGiven = NOT_GIVEN ,
479
479
mask : Iterable [float ] | NotGiven = NOT_GIVEN ,
480
480
num_steps : int | NotGiven = NOT_GIVEN ,
481
481
strength : float | NotGiven = NOT_GIVEN ,
@@ -935,7 +935,7 @@ async def run(
935
935
model_name : str ,
936
936
* ,
937
937
account_id : str ,
938
- image : Iterable [float ],
938
+ image : Union [ Iterable [float ], str ],
939
939
max_tokens : int | NotGiven = NOT_GIVEN ,
940
940
messages : Iterable [ai_run_params .ImageToTextMessage ] | NotGiven = NOT_GIVEN ,
941
941
prompt : str | NotGiven = NOT_GIVEN ,
@@ -991,7 +991,7 @@ async def run(
991
991
text : str | Union [str , List [str ]] | NotGiven = NOT_GIVEN ,
992
992
prompt : str | NotGiven = NOT_GIVEN ,
993
993
guidance : float | NotGiven = NOT_GIVEN ,
994
- image : Iterable [float ] | NotGiven = NOT_GIVEN ,
994
+ image : Iterable [float ] | Union [ Iterable [ float ], str ] | NotGiven = NOT_GIVEN ,
995
995
mask : Iterable [float ] | NotGiven = NOT_GIVEN ,
996
996
num_steps : int | NotGiven = NOT_GIVEN ,
997
997
strength : float | NotGiven = NOT_GIVEN ,
0 commit comments