Replaced the model by a float32 CPU model