ai-edge-torch-nightly 0.2.0.dev20240710__py3-none-any.whl → 0.2.0.dev20240712__py3-none-any.whl

ai_edge_torch/generative/utilities/stable_diffusion_loader.py

@@ -37,6 +37,9 @@ class ResidualBlockTensorNames:
 class AttentionBlockTensorNames:
   norm: str = None
   fused_qkv_proj: str = None
+  q_proj: str = None
+  k_proj: str = None
+  v_proj: str = None
   output_proj: str = None
 
 
@@ -106,12 +109,21 @@ def _map_to_converted_state(
     state_param: str,
     converted_state: Dict[str, torch.Tensor],
     converted_state_param: str,
+    squeeze_dims: bool = False,
 ):
   converted_state[f"{converted_state_param}.weight"] = state.pop(
       f"{state_param}.weight"
   )
+  if squeeze_dims:
+    converted_state[f"{converted_state_param}.weight"] = torch.squeeze(
+        converted_state[f"{converted_state_param}.weight"]
+    )
   if f"{state_param}.bias" in state:
     converted_state[f"{converted_state_param}.bias"] = state.pop(f"{state_param}.bias")
+    if squeeze_dims:
+      converted_state[f"{converted_state_param}.bias"] = torch.squeeze(
+          converted_state[f"{converted_state_param}.bias"]
+      )
 
 
 class BaseLoader(loader.ModelLoader):
@@ -179,17 +191,65 @@ class BaseLoader(loader.ModelLoader):
           f"{converted_state_param_prefix}.norm",
       )
     attention_layer_prefix = f"{converted_state_param_prefix}.attention"
-    _map_to_converted_state(
-        state,
-        tensor_names.fused_qkv_proj,
-        converted_state,
-        f"{attention_layer_prefix}.qkv_projection",
-    )
+    if tensor_names.fused_qkv_proj is not None:
+      _map_to_converted_state(
+          state,
+          tensor_names.fused_qkv_proj,
+          converted_state,
+          f"{attention_layer_prefix}.qkv_projection",
+      )
+    else:
+      _map_to_converted_state(
+          state,
+          tensor_names.q_proj,
+          converted_state,
+          f"{attention_layer_prefix}.q_projection",
+          squeeze_dims=True,
+      )
+      _map_to_converted_state(
+          state,
+          tensor_names.k_proj,
+          converted_state,
+          f"{attention_layer_prefix}.k_projection",
+          squeeze_dims=True,
+      )
+      _map_to_converted_state(
+          state,
+          tensor_names.v_proj,
+          converted_state,
+          f"{attention_layer_prefix}.v_projection",
+          squeeze_dims=True,
+      )
+      converted_state[f"{attention_layer_prefix}.qkv_projection.weight"] = torch.concat(
+          [
+              converted_state[f"{attention_layer_prefix}.q_projection.weight"],
+              converted_state[f"{attention_layer_prefix}.k_projection.weight"],
+              converted_state[f"{attention_layer_prefix}.v_projection.weight"],
+          ],
+          axis=0,
+      )
+      del converted_state[f"{attention_layer_prefix}.q_projection.weight"]
+      del converted_state[f"{attention_layer_prefix}.k_projection.weight"]
+      del converted_state[f"{attention_layer_prefix}.v_projection.weight"]
+      if config.attention_config.qkv_use_bias:
+        converted_state[f"{attention_layer_prefix}.qkv_projection.bias"] = torch.concat(
+            [
+                converted_state[f"{attention_layer_prefix}.q_projection.bias"],
+                converted_state[f"{attention_layer_prefix}.k_projection.bias"],
+                converted_state[f"{attention_layer_prefix}.v_projection.bias"],
+            ],
+            axis=0,
+        )
+        del converted_state[f"{attention_layer_prefix}.q_projection.bias"]
+        del converted_state[f"{attention_layer_prefix}.k_projection.bias"]
+        del converted_state[f"{attention_layer_prefix}.v_projection.bias"]
+
     _map_to_converted_state(
         state,
         tensor_names.output_proj,
         converted_state,
         f"{attention_layer_prefix}.output_projection",
+        squeeze_dims=True,
     )
 
   def _map_cross_attention_block(
@@ -269,7 +329,7 @@ class BaseLoader(loader.ModelLoader):
       converted_state: Dict[str, torch.Tensor],
       tensor_names: TransformerBlockTensorNames,
       converted_state_param_prefix: str,
-      config: unet_config.TransformerBlock2Dconfig,
+      config: unet_config.TransformerBlock2DConfig,
   ):
     _map_to_converted_state(
         state,
@@ -482,6 +542,10 @@ class BaseLoader(loader.ModelLoader):
       )
 
 
+# Alias class name for better code reading.
+ClipModelLoader = BaseLoader
+
+
 class AutoEncoderModelLoader(BaseLoader):
 
   @dataclass
@@ -668,7 +732,7 @@ class DiffusionModelLoader(BaseLoader):
                 stride=2,
                 padding=config.downsample_padding,
             ),
-            transformer_block_config=unet_config.TransformerBlock2Dconfig(
+            transformer_block_config=unet_config.TransformerBlock2DConfig(
                 attention_block_config=unet_config.AttentionBlock2DConfig(
                     dim=output_channel,
                     normalization_config=config.transformer_norm_config,
@@ -726,7 +790,7 @@ class DiffusionModelLoader(BaseLoader):
         ),
         num_layers=config.mid_block_layers,
         time_embedding_channels=config.time_embedding_blocks_dim,
-        transformer_block_config=unet_config.TransformerBlock2Dconfig(
+        transformer_block_config=unet_config.TransformerBlock2DConfig(
             attention_block_config=unet_config.AttentionBlock2DConfig(
                 dim=mid_block_channels,
                 normalization_config=config.transformer_norm_config,
@@ -789,7 +853,7 @@ class DiffusionModelLoader(BaseLoader):
                 mode=unet_config.SamplingType.NEAREST,
                 scale_factor=2,
             ),
-            transformer_block_config=unet_config.TransformerBlock2Dconfig(
+            transformer_block_config=unet_config.TransformerBlock2DConfig(
                 attention_block_config=unet_config.AttentionBlock2DConfig(
                     dim=output_channel,
                     normalization_config=config.transformer_norm_config,

ai_edge_torch/model.py

@@ -33,7 +33,10 @@ class Model(abc.ABC):
 
   @abc.abstractmethod
   def __call__(
-      self, *args: npt.ArrayLike, signature_name: str = cutils.DEFAULT_SIGNATURE_NAME
+      self,
+      *args: npt.ArrayLike,
+      signature_name: str = cutils.DEFAULT_SIGNATURE_NAME,
+      **kwargs,
   ) -> npt.ArrayLike | tuple[npt.ArrayLike]:
     raise NotImplementedError()
 
@@ -62,12 +65,16 @@ class TfLiteModel(Model):
     self._tflite_model = tflite_model
 
   def __call__(
-      self, *args: npt.ArrayLike, signature_name: str = cutils.DEFAULT_SIGNATURE_NAME
+      self,
+      *args: npt.ArrayLike,
+      signature_name: str = cutils.DEFAULT_SIGNATURE_NAME,
+      **kwargs,
   ) -> npt.ArrayLike | tuple[npt.ArrayLike]:
     """Runs inference on the edge model using the provided arguments.
 
     Args:
       *args: The arguments to be passed to the model for inference.
+      **kwargs: The arguments with specific names to be passed to the model for inference.
       signature_name: The name of the signature to be used for inference.
         The default signature is used if not provided.
     """
@@ -90,13 +97,14 @@ class TfLiteModel(Model):
       else:
         raise exception
 
-    if len(signature_list[signature_name]['inputs']) != len(args):
+    if len(signature_list[signature_name]['inputs']) != len(args) + len(kwargs):
       raise ValueError(
           f"The model requires {len(signature_list[signature_name]['inputs'])} arguments but {len(args)} was provided."
       )
 
     # Gather the input dictionary based on the signature.
     inputs = {f'args_{idx}': args[idx] for idx in range(len(args))}
+    inputs = {**inputs, **kwargs}
     outputs = runner(**inputs)
 
     return (

ai_edge_torch/testing/model_coverage/model_coverage.py

@@ -60,7 +60,8 @@ def _torch_tensors_to_np(*argv):
 def compare_tflite_torch(
     edge_model: Model,
     torch_eval_func: Callable,
-    input_data=None,
+    args=None,
+    kwargs=None,
     *,
     num_valid_inputs: int = 1,
     signature_name: str = None,
@@ -71,8 +72,9 @@ def compare_tflite_torch(
   Args:
     edge_model: Serialized ai_edge_torch.model.Model object.
     torch_eval_func: Callable function to evaluate torch model.
-    input_data: torch.tensor array  or a callable to generate a torch.tensor array
+    args: torch.tensor array or a callable to generate a torch.tensor array
       with random data, to pass into models during inference. (default None).
+    kwargs: dict of str to torch.tensor, or a callable to generate such.
     num_valid_inputs: Defines the number of times the random inputs will be generated (if a callable is provided for input_data).
     signature_name: If provided, specifies the name for the signature of the edge_model to run.
       Calls the default signature if not provided.
@@ -86,29 +88,33 @@ def compare_tflite_torch(
   # The supplied model_def.forward_args() will be executed num_valid_inputs
   # times to generate num_valid_inputs random inputs.
   torch_inputs = [
-      input_data() if callable(input_data) else input_data
+      (
+          (args() if callable(args) else args) or tuple(),
+          (kwargs() if callable(kwargs) else kwargs) or {},
+      )
       for _ in range(num_valid_inputs)
   ]
-  torch_outputs = [torch_eval_func(*xs) for xs in torch_inputs]
-  np_inputs = [_torch_tensors_to_np(xs) for xs in torch_inputs]
+  torch_outputs = [torch_eval_func(*args, **kwargs) for args, kwargs in torch_inputs]
+  np_inputs = [
+      (_torch_tensors_to_np(args), _torch_tensors_to_np(kwargs))
+      for args, kwargs in torch_inputs
+  ]
   np_outputs = [_torch_tensors_to_np(_flatten(ys)) for ys in torch_outputs]
 
   # Define inline utility function used throughout the function.
   def equal_fn(actual, expected):
     return np.allclose(actual, expected, atol=atol, rtol=rtol)
 
-  def get_actual_fn(input):
+  def get_edge_output(inputs):
+    args, kwargs = inputs
     if signature_name is None:
-      return _flatten(edge_model(*input))
+      return _flatten(edge_model(*args, **kwargs))
     else:
-      return _flatten(edge_model(*input, signature_name=signature_name))
-
-  def get_expected_fn(input=None, idx=0):
-    return np_outputs[idx]
+      return _flatten(edge_model(*args, **kwargs, signature_name=signature_name))
 
   for idx, np_input in enumerate(np_inputs):
-    output = get_actual_fn(np_input)
-    golden_output = get_expected_fn(np_input, idx)
+    output = get_edge_output(np_input)
+    golden_output = np_outputs[idx]
 
     is_output_len_eq = len(golden_output) == len(output)
 

{ai_edge_torch_nightly-0.2.0.dev20240710.dist-info → ai_edge_torch_nightly-0.2.0.dev20240712.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-edge-torch-nightly
-Version: 0.2.0.dev20240710
+Version: 0.2.0.dev20240712
 Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
 Home-page: https://github.com/google-ai-edge/ai-edge-torch
 Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI

{ai_edge_torch_nightly-0.2.0.dev20240710.dist-info → ai_edge_torch_nightly-0.2.0.dev20240712.dist-info}/RECORD

@@ -1,9 +1,9 @@
 ai_edge_torch/__init__.py,sha256=CNDboRP4zQBpz2hznNCQWcQCARvNXUm3DMa1Dw_XXFg,1067
-ai_edge_torch/model.py,sha256=kmcgELjsYl8YzF8nUF6P7q4i8MWS-pLGpfsy-yTUXmE,4243
+ai_edge_torch/model.py,sha256=8Ba9ia7TCM_fciulw6qObmzdcxL3IaLQKDqpR7Lxp-Q,4440
 ai_edge_torch/convert/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/convert/conversion.py,sha256=8K8jQuaCjlUWoj7jiimxp_zpN6mYThLOcQ858UDcYnE,4159
-ai_edge_torch/convert/conversion_utils.py,sha256=9BqCL38DErv1vEVGtT3BIJVhdwZjw2EQ-_m5UpvVVYE,11294
-ai_edge_torch/convert/converter.py,sha256=bjj5TV5_g4sGyuSh8ThEDydlNMqhkGSY4SzXK6vwhqI,6927
+ai_edge_torch/convert/conversion.py,sha256=StJHglvx6cii36oi8sj-tZda009e9UqR6ufZOZkP1SY,4137
+ai_edge_torch/convert/conversion_utils.py,sha256=PKXIlSCU-8DhppNBh9ICDNUlEOpV0HgCbt85jDVe3rA,13394
+ai_edge_torch/convert/converter.py,sha256=hSrW6A-kix9cjdD6CuLL7rseWrLKoV6GRy-iUSW_nZc,7875
 ai_edge_torch/convert/to_channel_last_io.py,sha256=zo5tY3yDhY_EPCkrL1XSXs2uRFS8B4_qu08dSjNsUGk,2778
 ai_edge_torch/convert/fx_passes/__init__.py,sha256=EPs4PSIDLuRH5EBETi6deaOvaaf_Q4xD3_9NVcR7x8o,2810
 ai_edge_torch/convert/fx_passes/_pass_base.py,sha256=ijVyDclPnd6a0DWWUJkwR4igj6f82S-cE1-83QGPvgw,1652
@@ -22,7 +22,7 @@ ai_edge_torch/convert/fx_passes/optimize_layout_transposes_pass/layout_partition
 ai_edge_torch/convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/greedy.py,sha256=8uHJbIwPMTgeSfYVba163pkXSQkHLxFwar_8A1AhgAM,2279
 ai_edge_torch/convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/min_cut.py,sha256=lklGxE1R32vsjFbhLLBDEFL4pfLi_iTgI9Ftb6Grezk,7156
 ai_edge_torch/convert/test/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/convert/test/test_convert.py,sha256=2qPmmGqnfV_o1gfsSdjGq3-JR1b323ligiy5MdAv9NA,8021
+ai_edge_torch/convert/test/test_convert.py,sha256=h0vOffr8saDQRkiXljNWDZ17EBjnS4xAtxd8DxETleY,9081
 ai_edge_torch/convert/test/test_convert_composites.py,sha256=_Ojc-H6GOS5s8ek3_8eRBL_AiCs-k3srziPJ2R4Ulrg,7255
 ai_edge_torch/convert/test/test_convert_multisig.py,sha256=kMaGnHe9ylfyU68qCifYcaGwJqyejKz--QQt9jS2oUA,4537
 ai_edge_torch/convert/test/test_to_channel_last_io.py,sha256=I8c4ZG3v1vo0yxQYzLK_BTId4AOL9vadHGDtfCUZ4UI,2930
@@ -43,12 +43,12 @@ ai_edge_torch/generative/examples/phi2/convert_to_tflite.py,sha256=uF1A2EX8xYie3
 ai_edge_torch/generative/examples/phi2/phi2.py,sha256=PMhKC6JCAMYSj2F3UmWHWK4rTcXD-B6PuehaoDccRqk,5562
 ai_edge_torch/generative/examples/stable_diffusion/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/examples/stable_diffusion/attention.py,sha256=Lo4Dq7a3Kg-lyH56iqGtqCo5UaClQHRCTDdNagXGTo8,3535
-ai_edge_torch/generative/examples/stable_diffusion/clip.py,sha256=qU1wVEcn_biwCuDguZljhlLGzpLIqgqC31Dh_lXquQc,3720
-ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py,sha256=wVEjsKd5JCIiYf5GF19rOXs2NHscZh0D69mxaS4f0Sk,4182
-ai_edge_torch/generative/examples/stable_diffusion/decoder.py,sha256=RgxedILk7iNMb0mhE4VkCs6d7BnFzYhR3vspUkC0-1o,11425
-ai_edge_torch/generative/examples/stable_diffusion/diffusion.py,sha256=sRevfsmCun7zbceJbOstLKNUsLwzQDsGm7Mi2JmlREg,26021
+ai_edge_torch/generative/examples/stable_diffusion/clip.py,sha256=P-cUUQaQKGKV2p-7hvLJ--RpCIA7gk8WCDRgg0pNtd0,4331
+ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py,sha256=XwV1z7cVkQ947k_ERftEeL8n0NUFCJAltLtqDVfzYGI,4704
+ai_edge_torch/generative/examples/stable_diffusion/decoder.py,sha256=xHcmOZaW7hoWlEEEqtB4FWoHMw5AsGHPHXMNiXEfviY,13814
+ai_edge_torch/generative/examples/stable_diffusion/diffusion.py,sha256=G-MgiEM_PpegNMePBPuNQDeUfjk42EYrVZAyJHC54AY,28468
 ai_edge_torch/generative/examples/stable_diffusion/encoder.py,sha256=mgbxkeFDMkNIGmnbcFTIFPu8EWKokghiviYIOB2lE3Q,3437
-ai_edge_torch/generative/examples/stable_diffusion/pipeline.py,sha256=FCbnwlkpYYb-tF7KscbSYjNEdg7XnuLju1cDuIRoQv8,8277
+ai_edge_torch/generative/examples/stable_diffusion/pipeline.py,sha256=AopJ-KE74lzq4QJUP_hYeiXvGth7uWv7nNKqkhtcoF8,8277
 ai_edge_torch/generative/examples/stable_diffusion/tokenizer.py,sha256=r9RqbyNvuvXOGu3ojtl7ZmbC7o4Pt8aUKAhN1yCdtEc,3397
 ai_edge_torch/generative/examples/stable_diffusion/util.py,sha256=NFpOfA4KN0JpShm5QvuYbQYZ844NzexWD8nV3WjMOZM,2397
 ai_edge_torch/generative/examples/stable_diffusion/samplers/__init__.py,sha256=uQWKzCD_49ackNFrt50H04dkDXxfAwUCtMWWQre5SVE,830
@@ -59,7 +59,7 @@ ai_edge_torch/generative/examples/stable_diffusion/samplers/sampler.py,sha256=5i
 ai_edge_torch/generative/examples/t5/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/examples/t5/convert_to_tflite.py,sha256=7RwaZQaKhFt3zKAUbFjq95CSYhL1nd9BVSbSRNJp4-4,4529
 ai_edge_torch/generative/examples/t5/t5.py,sha256=L6YrVzUEzP-Imb8W28LdukFGrx1aWSzz1kyYK_9RFZM,21087
-ai_edge_torch/generative/examples/t5/t5_attention.py,sha256=rkMwi-NJGBXHm5S57Rsj1LbcoVdyRkS7GmIBuU6F_2E,8274
+ai_edge_torch/generative/examples/t5/t5_attention.py,sha256=KaGzCAViNOpJIQbRF-ItouuVPqI9nroWRRGN-KFYKZs,8357
 ai_edge_torch/generative/examples/test_models/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/examples/test_models/toy_model.py,sha256=Sf3ZMYv-iuMRKAKLow47qth8vTF1zl6i8TxJ9uT_StU,3885
 ai_edge_torch/generative/examples/test_models/toy_model_with_external_kv_cache.py,sha256=zwCmCnhr-vhBwHqv9i7xMasdBGVNqAGxZvWsncsJn58,5543
@@ -68,21 +68,21 @@ ai_edge_torch/generative/examples/tiny_llama/__init__.py,sha256=47DEQpj8HBSa-_TI
 ai_edge_torch/generative/examples/tiny_llama/convert_to_tflite.py,sha256=nT7Fh-f5ZdwaK3dPoCvZflpJ4fRHjLdFMjk1_uw3-b8,2559
 ai_edge_torch/generative/examples/tiny_llama/tiny_llama.py,sha256=IFRLPG9wz_aLl_zV_6CETCjSM03ukA6bZqqyDLVACuw,5651
 ai_edge_torch/generative/fx_passes/__init__.py,sha256=aXvYiaHDvETIrh0Q9DDZA_ZBiazGk80DT6nt7lLtC1o,1172
-ai_edge_torch/generative/fx_passes/remove_sdpa_zero_mask_pass.py,sha256=IehLwFNwa0C9fnk1pmNmyfuAwwWbuwdyKy46BSqNVdI,1948
+ai_edge_torch/generative/fx_passes/remove_sdpa_zero_mask_pass.py,sha256=BCAcc_OcEjvbaXQSbc8vlKeMad7E3gCA4BNsUdWRwBI,1966
 ai_edge_torch/generative/layers/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/layers/attention.py,sha256=AW0Qo3uOIe6p1rJNJ6zR_r4fqL2y-6QJHh0yUd-5Yb0,11966
 ai_edge_torch/generative/layers/attention_utils.py,sha256=hXhuyKblPPxKIRzlAf1YNlwHgpbj-6nReRLhRHELx5k,6350
 ai_edge_torch/generative/layers/builder.py,sha256=jAyrR5hsSI0aimKZumyvxdJ1GovERIfsK0g-dezX2gs,4163
 ai_edge_torch/generative/layers/feed_forward.py,sha256=4j2QaSCw59Jkk_ixKDpKEj7FLRauzuExTiSNRzAjAhE,2820
 ai_edge_torch/generative/layers/kv_cache.py,sha256=4uiZLO3om5G3--kT04Jt0esEYznbkJ7QLzSHfb8mjc4,3090
-ai_edge_torch/generative/layers/model_config.py,sha256=aQLtOPdGpehfnb4aGO-iILLAsRU5t7j6opyezPEUY_w,4673
+ai_edge_torch/generative/layers/model_config.py,sha256=s6aIBib_LhjZC3p1pRxjcg3mf1BUrGqPQdsb6G83U-c,5028
 ai_edge_torch/generative/layers/normalization.py,sha256=M27eW3TcNK20oaXClXtfnu0lLWrAGrSKSsbegRWnj3c,1867
 ai_edge_torch/generative/layers/rotary_position_embedding.py,sha256=12SsCuoRuLNCwnFGe_pHDOZEBwBcqXs87Aj0PaWWw4E,1383
 ai_edge_torch/generative/layers/scaled_dot_product_attention.py,sha256=dYafGC205QE5CLIbBTCI-7eVvEGZEHzs1toPEhemeDs,3391
 ai_edge_torch/generative/layers/unet/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/layers/unet/blocks_2d.py,sha256=H45wsXA6iJi_Mjd66NiQrh7i1fx05r9o_FI-fSnhVts,26538
+ai_edge_torch/generative/layers/unet/blocks_2d.py,sha256=T70veX57CC9uNidwzoVGzOu-CwzcYMBr1Zk_0bq5UlM,26538
 ai_edge_torch/generative/layers/unet/builder.py,sha256=NmJiZ2-e1wbv9jnvI3VCyUJlONV5ZAOz-RTc7ipAZ5U,1872
-ai_edge_torch/generative/layers/unet/model_config.py,sha256=FrIO-CR8aRIV2i8aFqom_4S7WCEDLMyYwo6U0oFyn7A,9097
+ai_edge_torch/generative/layers/unet/model_config.py,sha256=GU12QEJwO6ukveMR9JRsrhE0YIPKuhk1U81CylmOQTA,9097
 ai_edge_torch/generative/quantize/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/quantize/example.py,sha256=Oy-Ss1oKXMu5RVOGt8QiUwKtrHEfhbVjTXXjxPcOqDA,1536
 ai_edge_torch/generative/quantize/quant_attrs.py,sha256=n1Fm8BFC8gJa_oiwwAOOghJyHtOXYZ4q-5ZRy4pHrIw,1957
@@ -97,8 +97,8 @@ ai_edge_torch/generative/test/loader_test.py,sha256=N88CbrLW7Q2x1EyurwdXQ6YjsA-y
 ai_edge_torch/generative/test/test_model_conversion.py,sha256=LsPTrLC1I4JW2GowTS3V9Eu257vLHr2Yj5f_qaFUX84,7589
 ai_edge_torch/generative/test/test_quantize.py,sha256=TxZwe2cCTfwq9t2thBuYiLdp5Xu2cspCbQgziZ3Oo7k,5269
 ai_edge_torch/generative/utilities/__init__.py,sha256=-_jxnnFnCgnTU4oTm4MnRsvL5lqhomBNdFBbqfmfHPo,720
-ai_edge_torch/generative/utilities/loader.py,sha256=Hs92478j1g4jQGvbdP1aWvOy907HjwqQZE-NFy6HELo,11326
-ai_edge_torch/generative/utilities/stable_diffusion_loader.py,sha256=7ChqrnthD7I-Be6vkRvYTRhbGQ3tqMbikLpjY5HpSzE,30890
+ai_edge_torch/generative/utilities/loader.py,sha256=NTaCrU2qmeJpqdAau13ZgyeOpwATqhZB68GY0LZjU6A,11438
+ai_edge_torch/generative/utilities/stable_diffusion_loader.py,sha256=zixjZryUaCSDKmfPkQvYwbPJhUyTmZ4AK_lWN8iFo68,33324
 ai_edge_torch/generative/utilities/t5_loader.py,sha256=h1FQzt4x8wiQMX4NzYNVIaJGLr_YKH0sojBvy0amexM,16503
 ai_edge_torch/hlfb/__init__.py,sha256=rrje8a2iuKboBoV96bVq7nlS9HsnuEMbHE5JiWmCxFA,752
 ai_edge_torch/hlfb/mark_pattern/__init__.py,sha256=2VXnHcGf23VOuP-1GriGIpuL98leBB8twp_qaScMnmc,4799
@@ -113,9 +113,9 @@ ai_edge_torch/quantize/pt2e_quantizer_utils.py,sha256=yjzKoptnfEeW_sN7sODUfj3nCt
 ai_edge_torch/quantize/quant_config.py,sha256=eO9Ra160ITjQSyRBEGy6nNIVH3gYacSWDdN5XtvHwjc,3148
 ai_edge_torch/testing/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
-ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=EIyKz-HY70DguWuSrJal8LpYXQ5ZSEUf3ZrVl7jikFM,4286
-ai_edge_torch_nightly-0.2.0.dev20240710.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-ai_edge_torch_nightly-0.2.0.dev20240710.dist-info/METADATA,sha256=6ask_HCsla1Tzx5_ORpPGrdvtwYAwS6BB3jNV31Jo9g,1745
-ai_edge_torch_nightly-0.2.0.dev20240710.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-ai_edge_torch_nightly-0.2.0.dev20240710.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
-ai_edge_torch_nightly-0.2.0.dev20240710.dist-info/RECORD,,
+ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=kzIulTldq8R9E-lAZsvfSTvLu3FYEX7b9DyYM3qisXM,4485
+ai_edge_torch_nightly-0.2.0.dev20240712.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+ai_edge_torch_nightly-0.2.0.dev20240712.dist-info/METADATA,sha256=BGHmRLYo3ko7KRysDP59YexTpPn45jrtpRwqQkPAM5s,1745
+ai_edge_torch_nightly-0.2.0.dev20240712.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+ai_edge_torch_nightly-0.2.0.dev20240712.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
+ai_edge_torch_nightly-0.2.0.dev20240712.dist-info/RECORD,,