PrunaAI

/

internlm-internlm2-math-7b-QUANTO-int2bit-smashed

like 0

Follow

Pruna AI 154

• "torch.nn.modules.sparse.Embedding",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.configuration_internlm2.InternLM2Config",
• "transformers.generation.configuration_utils.GenerationConfig",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.modeling_internlm2.InternLM2RotaryEmbedding",
• "collections.OrderedDict",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.modeling_internlm2.InternLM2RMSNorm",
• "__builtin__.set",
• "torch.bfloat16",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.modeling_internlm2.InternLM2Model",
• "torch.BFloat16Storage",
• "quanto.nn.qlinear.QLinear",
• "torch._utils._rebuild_tensor_v2",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.modeling_internlm2.InternLM2Attention",
• "quanto.tensor.qtype.qtype",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.modeling_internlm2.InternLM2MLP",
• "torch.int8",
• "torch._utils._rebuild_parameter",
• "torch.FloatStorage",
• "torch.nn.modules.activation.SiLU",
• "torch.nn.modules.container.ModuleList",
• "torch.device",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.modeling_internlm2.InternLM2DecoderLayer",
• "transformers_modules.internlm.internlm2-math-7b.2e503ece6bb48495255bb53e495f81196c424985.modeling_internlm2.InternLM2ForCausalLM"

How to fix it?