code: def __init__(
      self,
      image_H,
      image_W,
      fake_len=100000,
      image_key="images",
      txt_key="txt",
      seq_len=80,
      context_dim=768,
  ):
      super().__init__()
      self.fake_len = fake_len
      self.H = image_H
      self.W = image_W
      self.image_key = image_key
      self.txt_key = txt_key
      assert image_key.endswith("encoded") == txt_key.endswith(
          "encoded"
      ), 'In precached mode, first and second stage key must both end with "encoded"'
      self.precached = self.image_key.endswith("encoded")
      self.seq_len = seq_len
      self.context_dim = context_dim
  
file_path: NeMo/nemo/collections/multimodal/data/stable_diffusion/stable_diffusion_dataset.py
imports: import torch
  from nemo.collections.multimodal.data.common.webdataset import WebDatasetCommon
  from nemo.collections.multimodal.data.stable_diffusion.augmentation.augmentations import construct_image_augmentations, identical_transform
  from nemo.core.classes import Dataset
  from nemo.utils import logging
label_desc: Use image node_id 1 to create a new SDSyntheticDataset object from inherited base classes: NeMoDataset with example: obj = SDSyntheticDataset(image_H, image_W, fake_len, image_key, txt_key, seq_len, context_dim)