WeReCooking2

/

FE2E-INT8

like 0

Follow

We are cooking Two 2

• "modules.layers.QKNorm",
• "modules.layers.TimestepEmbedder",
• "torch.ScriptObject",
• "torch.nn.modules.normalization.LayerNorm",
• "torch.nn.modules.linear.Identity",
• "__builtin__.set",
• "modules.layers.EmbedND",
• "torch.QInt8Storage",
• "modules.layers.DoubleStreamBlock",
• "modules.layers.SelfAttention",
• "torch.nn.modules.dropout.Dropout",
• "torch.FloatStorage",
• "torch.nn.modules.activation.GELU",
• "torch._utils._rebuild_parameter",
• "modules.layers.RMSNorm",
• "modules.connector_edit.IndividualTokenRefinerBlock",
• "collections.OrderedDict",
• "modules.layers.MLP",
• "modules.layers.LastLayer",
• "torch.nn.modules.container.Sequential",
• "torch.nn.modules.container.ModuleList",
• "modules.connector_edit.Qwen2Connector",
• "torch.ao.nn.quantized.modules.linear.LinearPackedParams",
• "torch.qint8",
• "torch.per_tensor_affine",
• "modules.layers.TextProjection",
• "modules.connector_edit.SingleTokenRefiner",
• "modules.connector_edit.IndividualTokenRefiner",
• "torch.ao.nn.quantized.dynamic.modules.linear.Linear",
• "modules.layers.Modulation",
• "torch._utils._rebuild_qtensor",
• "torch._utils._rebuild_tensor_v2",
• "modules.layers.MLPEmbedder",
• "modules.model_edit.Step1XParams",
• "modules.layers.SingleStreamBlock",
• "modules.model_edit.Step1XEdit",
• "torch.nn.modules.activation.SiLU"

How to fix it?

• "torch.nn.modules.conv.Conv2d",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch.nn.modules.normalization.GroupNorm",
• "torch.nn.modules.container.ModuleList",
• "modules.autoencoder.Downsample",
• "modules.autoencoder.AttnBlock",
• "__builtin__.set",
• "torch.nn.modules.module.Module",
• "modules.autoencoder.Decoder",
• "modules.autoencoder.DiagonalGaussian",
• "modules.autoencoder.ResnetBlock",
• "torch._utils._rebuild_tensor_v2",
• "modules.autoencoder.Encoder",
• "modules.autoencoder.AutoEncoder",
• "torch._utils._rebuild_parameter",
• "modules.autoencoder.Upsample"

How to fix it?