[BUG]: 使用带有 Gemini 插件的 CPUAdam 时,出现 `AssertionError: div_scale 应保持默认值`

import json import os import random import sys import torch import torch.nn as nn import torch.optim as optim from torch.utils.data import Dataset, DataLoader import colossalai from colossalai.booster import Booster from colossalai.nn.optimizer import CPUAdam, Lamb from colossalai.booster.plugin import GeminiPlugin