FIX Missing device map for facebook/opt-125m (#2675)

Fixes the failing EETQ test in the nighly multi device CI.

In #2612, fixed device_maps were added for multi-GPU training as we
could not rely on device_map="auto". While doing this change, one
device_map was missing, namely for facebook/opt-125m, which is used in
the EETQ multi device test. This device_map was now added. This makes
the test pass locally.

Author: BenjaminBossan

tests/test_gpu_examples.py

@@ -143,6 +143,24 @@
         "model.decoder.layers.31": 1,
         "lm_head": 0,  # tied with embed_tokens
     },
+    "facebook/opt-125m": {
+        "model.decoder.embed_tokens": 0,
+        "model.decoder.embed_positions": 0,
+        "model.decoder.final_layer_norm": 1,
+        "model.decoder.layers.0": 0,
+        "model.decoder.layers.1": 0,
+        "model.decoder.layers.2": 0,
+        "model.decoder.layers.3": 0,
+        "model.decoder.layers.4": 0,
+        "model.decoder.layers.5": 0,
+        "model.decoder.layers.6": 1,
+        "model.decoder.layers.7": 1,
+        "model.decoder.layers.8": 1,
+        "model.decoder.layers.9": 1,
+        "model.decoder.layers.10": 1,
+        "model.decoder.layers.11": 1,
+        "lm_head": 0,
+    },
     "marcsun13/opt-350m-gptq-4bit": {
         "model.decoder.embed_tokens": 0,
         "model.decoder.embed_positions": 0,