Checking what ESP8266 module that is in use

Question

I have this script where I have connected an ESP-01 to an SSD1306 OLED screen. However, the script I'm using needs adaption for ESP-01 as this module requires the I2C pins to be defined.

So basically I need to add a line that says Wire.begin(2, 0); so make the screen work with the ESP-01.

Whoever, I would love to have a check that is wrapped in a #ifdef so the code will work out of the box with multiple ESP modules without changing the source code.

Is there some way I can check if the module in use is an ESP-01?