fix: state dict call.

convert.py

@@ -71,7 +71,7 @@ def run_conversion(text_encoder_weights: str = None, unet_weights: str = None):
         text_encoder_state_dict_from_pt = pt_text_encoder.state_dict()
         run_assertion(text_encoder_state_dict_from_pt, text_encoder_state_dict_from_tf)
     if unet_weights is None:
-        unet_state_dict_from_pt = pt_text_encoder.state_dict()
+        unet_state_dict_from_pt = pt_unet.state_dict()
         run_assertion(unet_state_dict_from_pt, unet_state_dict_from_tf)
 
     print(