{
    "model.embed_tokens": "cuda:0",
    "model.layers.0": "cuda:0",
    "model.layers.1": "cuda:0",
    "model.layers.2": "cuda:0",
    "model.layers.3": "cuda:0",
    "model.layers.4": "cuda:0",
    "model.layers.5": "cuda:0",
    "model.layers.6": "cuda:0",
    "model.layers.7": "cuda:1",
    "model.layers.8": "cuda:1",
    "model.layers.9": "cuda:1",
    "model.layers.10": "cuda:1",
    "model.layers.11": "cuda:1",
    "model.layers.12": "cuda:1",
    "model.layers.13": "cuda:1",
    "model.layers.14": "cuda:2",
    "model.layers.15": "cuda:2",
    "model.layers.16": "cuda:2",
    "model.layers.17": "cuda:2",
    "model.layers.18": "cuda:2",
    "model.layers.19": "cuda:2",
    "model.layers.20": "cuda:2",
    "model.layers.21": "cuda:3",
    "model.layers.22": "cuda:3",
    "model.layers.23": "cuda:3",
    "model.layers.24": "cuda:3",
    "model.layers.25": "cuda:3",
    "model.layers.26": "cuda:3",
    "model.layers.27": "cuda:3",
    "model.layers.28": "cuda:4",
    "model.layers.29": "cuda:4",
    "model.layers.30": "cuda:4",
    "model.layers.31": "cuda:4",
    "model.layers.32": "cuda:4",
    "model.layers.33": "cuda:4",
    "model.layers.34": "cuda:4",
    "model.layers.35": "cuda:4",
    "model.layers.36": "cuda:5",
    "model.layers.37": "cuda:5",
    "model.layers.38": "cuda:5",
    "model.layers.39": "cuda:5",
    "model.layers.40": "cuda:5",
    "model.layers.41": "cuda:5",
    "model.layers.42": "cuda:5",
    "model.layers.43": "cuda:5",
    "model.layers.44": "cuda:6",
    "model.layers.45": "cuda:6",
    "model.layers.46": "cuda:6",
    "model.layers.47": "cuda:6",
    "model.layers.48": "cuda:6",
    "model.layers.49": "cuda:6",
    "model.layers.50": "cuda:6",
    "model.layers.51": "cuda:6",
    "model.layers.52": "cuda:7",
    "model.layers.53": "cuda:7",
    "model.layers.54": "cuda:7",
    "model.layers.55": "cuda:7",
    "model.layers.56": "cuda:7",
    "model.layers.57": "cuda:7",
    "model.layers.58": "cuda:7",
    "model.layers.59": "cuda:7",
    "model.norm": "cuda:7",
    "lm_head": "cuda:7"
}
