class_docstr: FID model.
class_docstr_tok: ["FID","model","."]
code: def build_flax_model(self) -> nn.Module:
      return FusionInDecoderSoftModule(self.config.model)
  
imports: from typing import Any, Dict, Optional
  import flax.linen
  import jax
  import jax.numpy
  import ml_collections
  from scenic.model_lib.base_models import base_model
  from scenic.projects.knowledge_visual_language.models import constants
  from scenic.projects.knowledge_visual_language.models import layers
  from scenic.projects.knowledge_visual_language.models import losses
  from scenic.projects.knowledge_visual_language.models import metrics
  from scenic.projects.knowledge_visual_language.models import vit
  from scenic.projects.t5 import layers
  from scenic.projects.t5 import model
label_desc: Use image node_id 1 for calling the FIDSoftModel obj's underlying member method code with example usage: obj.build_flax_model() and returns: FusionInDecoderSoftModule