  from lib.model.nn_blocks import (Conv2DOutput, Conv2DBlock, ResidualBlock, SeparableConv2DBlock,
                                   UpscaleBlock)
  from .original import Model as OriginalModel, KerasModel
  
  if get_backend() == "amd":
      from keras.initializers import RandomNormal
      from keras.layers import add, Dense, Flatten, Input, LeakyReLU, Reshape
 
</s> remove from keras.initializers import RandomNormal
from keras.layers import add, Dense, Flatten, Input, LeakyReLU, Reshape

 </s> add  </s> remove from .original import Model as OriginalModel, KerasModel
 </s> add from .original import Model as OriginalModel, KerasModel, Dense, Flatten, Input, Reshape

from lib.utils import get_backend

  from lib.model.nn_blocks import (Conv2DOutput, Conv2DBlock, ResidualBlock, SeparableConv2DBlock,
                                   UpscaleBlock)
from lib.utils import get_backend

  from .original import Model as OriginalModel, KerasModel
  
  if get_backend() == "amd":
      from keras.initializers import RandomNormal
      from keras.layers import add, Dense, Flatten, Input, LeakyReLU, Reshape
keep add keep keep keep keep keep
 Update all Keras Imports to be conditional (#1214)

* Remove custom keras importer

* first round keras imports fix

* launcher.py: Remove KerasFinder references

* 2nd round keras imports update (lib and extract)

* 3rd round keras imports update (train)

* remove KerasFinder from tests

* 4th round keras imports update (tests) 
plugins/train/model/villain.py