  from lib.utils import get_backend
  
  if get_backend() == "amd":
      from . import losses_plaid as losses
  else:
      from . import losses_tf as losses
 
</s> remove from lib.model import layers, normalization
 </s> add from lib.model import layers </s> remove     if isinstance(layer, (layers.ReflectionPadding2D, normalization.InstanceNormalization)):
        layer.build(input_shape)
 </s> add     layer.build(input_shape)
from .normalization import *
  from lib.utils import get_backend
  
from .normalization import *
  if get_backend() == "amd":
      from . import losses_plaid as losses
  else:
      from . import losses_tf as losses
keep add keep keep keep keep
 lib.model.normalization:
  - refactor
  - add LayerNorm
  - add RMSNorm 
lib/model/__init__.py