@@ -419,7 +419,7 @@ def run(
419419 model_name : str ,
420420 * ,
421421 account_id : str ,
422-  image : Iterable [float ],
422+  image : Union [ Iterable [float ],  str ],
423423 max_tokens : int  |  NotGiven  =  NOT_GIVEN ,
424424 messages : Iterable [ai_run_params .ImageToTextMessage ] |  NotGiven  =  NOT_GIVEN ,
425425 prompt : str  |  NotGiven  =  NOT_GIVEN ,
@@ -475,7 +475,7 @@ def run(
475475 text : str  |  Union [str , List [str ]] |  NotGiven  =  NOT_GIVEN ,
476476 prompt : str  |  NotGiven  =  NOT_GIVEN ,
477477 guidance : float  |  NotGiven  =  NOT_GIVEN ,
478-  image : Iterable [float ] |  NotGiven  =  NOT_GIVEN ,
478+  image : Iterable [float ] |  Union [ Iterable [ float ],  str ]  |   NotGiven  =  NOT_GIVEN ,
479479 mask : Iterable [float ] |  NotGiven  =  NOT_GIVEN ,
480480 num_steps : int  |  NotGiven  =  NOT_GIVEN ,
481481 strength : float  |  NotGiven  =  NOT_GIVEN ,
@@ -935,7 +935,7 @@ async def run(
935935 model_name : str ,
936936 * ,
937937 account_id : str ,
938-  image : Iterable [float ],
938+  image : Union [ Iterable [float ],  str ],
939939 max_tokens : int  |  NotGiven  =  NOT_GIVEN ,
940940 messages : Iterable [ai_run_params .ImageToTextMessage ] |  NotGiven  =  NOT_GIVEN ,
941941 prompt : str  |  NotGiven  =  NOT_GIVEN ,
@@ -991,7 +991,7 @@ async def run(
991991 text : str  |  Union [str , List [str ]] |  NotGiven  =  NOT_GIVEN ,
992992 prompt : str  |  NotGiven  =  NOT_GIVEN ,
993993 guidance : float  |  NotGiven  =  NOT_GIVEN ,
994-  image : Iterable [float ] |  NotGiven  =  NOT_GIVEN ,
994+  image : Iterable [float ] |  Union [ Iterable [ float ],  str ]  |   NotGiven  =  NOT_GIVEN ,
995995 mask : Iterable [float ] |  NotGiven  =  NOT_GIVEN ,
996996 num_steps : int  |  NotGiven  =  NOT_GIVEN ,
997997 strength : float  |  NotGiven  =  NOT_GIVEN ,
0 commit comments