+h=SrSSKJr SSKrSSKJr SSKrSSKJ r SSK J r SSK J r \ R"\5rSrS rg) z!PyTorch - Flax general utilities.)UnpicklingErrorN) from_bytes) flatten_dict)loggingc[US5n[SUR55nSSS5 [UW5$!,(df  N=f![an[U5nUR5R S5(a [ S5e[ Ue!,(df  SnAN|=f![[ 4a [SUS35ef=fSnAff=f)NrbversionzYou seem to have cloned a repository without having git-lfs installed. Please install git-lfs and run `git lfs install` followed by `git lfs pull` in the folder you cloned.zUnable to convert z to Flax deserializable object. ) openrreadr startswithOSError ValueErrorUnicodeDecodeErrorEnvironmentError"load_flax_weights_in_pytorch_model)pt_model model_file flax_state_f flax_stateefs f/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/models/modeling_pytorch_flax_utils.py%load_flax_checkpoint_in_pytorch_modelr%sf *d #|#D,*;*;*=>J$ .h CC!$ #  f fj!Q668&&y11!. %!+"!!#J/ f"%7 |Cc#de e f fsU A=A A A A C B0$7B B- %B0-B00 CCCc SSKn[ [ R RSU55R5n[U5(a6[RS5 [ R RSU5nSUl [ USS 9nUR5n/n[UR55nUR5GH upUR!S5n U S S :Xa1U R"S :Xa!U SS S /-n [$R&"U S5n O1U S S :XaU SS S /-n U R(n OU S S:Xa U SS S /-n SU ;a[+U 5HupU R-SS5R-SS5R-SS5R-SS5R-SS5R-SS5R-SS5R-SS 5R-S!S"5R-S#S$5X'M SR/U 5n X;aU R0X]R0:wa+[3S%US&X]R0S'U R0S35e[5U [6R85(d[6R:"U 5OU n UR=U 5X]'UR?U 5 GMURAU 5 GM URCU5 [EU5n[GU5S:a^[RS(URHRJS)US*URHRJS+URHRJS,3 5 [GU5S:a0[RS-URHRJS.US/35 U$![a [RS5 ef=f)0z(Load flax checkpoints in a PyTorch modelrNzLoading Flax weights in PyTorch requires both PyTorch and Flax to be installed. Please see https://pytorch.org/ and https://flax.readthedocs.io/en/latest/installation.html for installation instructions.c<UR[R:H$N)dtypejnpbfloat16)xs r4load_flax_weights_in_pytorch_model..HsCLLAXzFound ``bfloat16`` weights in Flax model. Casting all ``bfloat16`` weights to ``float32`` before loading those in PyTorch model.cUR[R:XaUR[R 5$U$r)rrr astypenpfloat32)paramss rr"r#Rs)  8T6==4`Z``r$.)sepkernelweight)rrscaletime_embedding_0z.0_1z.1_2z.2_3z.3_4z.4_5z.5_6z.6_7z.7_8z.8_9z.9z.Flax checkpoint seems to be incorrect. Weight z was expected to be of shape z , but is zQSome weights of the Flax model were not used when initializing the PyTorch model z: z, - This IS expected if you are initializing z from a Flax model trained on another task or with another architecture (e.g. initializing a BertForSequenceClassification model from a FlaxBertForPreTraining model). - This IS NOT expected if you are initializing z from a Flax model that you expect to be exactly identical (e.g. initializing a BertForSequenceClassification model from a FlaxBertForSequenceClassification model).zSome weights of zE were not initialized from the Flax model and are newly initialized: zo You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.)&torch ImportErrorloggererrorrjax tree_utiltree_mapvaluesanywarningbase_model_prefix state_dictsetkeysitemssplitndimr transposeT enumeratereplacejoinshaper isinstancer'ndarrayasarray from_numpyremoveappendload_state_dictlistlen __class____name__)rrr? is_type_bf16flax_state_dict pt_model_dictunexpected_keys missing_keysflax_key_tuple flax_tensorflax_key_tuple_arrayiflax_key_tuple_stringflax_keys rrr:s  6 67XZd efmmoL <  5 ]]++ `bl "$H":37O'')MO}))+,L'6'<'<'>#-33C8  #x /K4D4D4I#7#<z#I -- \BK !" % 1#7#<z#I %--K !" % 0#7#<z#I #7 7,56J,K()11$=WT4(WT4(WT4(WT4(WT4(WT4(WT4(WT4(WT4(%'-L8801  $  M$;$A$AA D^DTU&&3&=&C&C%DIkN_N_M``ac >H UWU_U_=`=`bjj5fq */*:*:;*G '##H-  " "8 ,W(?Z ]+ %L ?a ""++,B.?@""++,-88@8J8J8S8S7TU9  9  <1x11::;<)N+5 5 OE     s N++!O )__doc__picklerrC jax.numpynumpyrr'flax.serializationrflax.traverse_utilrutilsr get_loggerr`rArrr$rrus=(" )+   H %D*gr$