PrunaAI

/

llama-moe-LLaMA-MoE-v1-3_5B-2_8-QUANTO-float8bit-smashed

like 0

• "torch.float8_e4m3fn",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.TopKBalancedNoisyGate",
• "torch.nn.modules.activation.Softplus",
• "torch.device",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LlamaRMSNorm",
• "torch.nn.modules.sparse.Embedding",
• "torch._utils._rebuild_parameter",
• "torch.nn.modules.activation.SiLU",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LinearGLUExperts",
• "torch.bfloat16",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LlamaMoEDecoderLayer",
• "torch.nn.modules.activation.Tanh",
• "torch.nn.modules.container.ModuleList",
• "torch._utils._rebuild_tensor_v2",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LlamaMoEModel",
• "quanto.tensor.qtype.qtype",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LinearGLUMoELayer",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LlamaRotaryEmbedding",
• "torch.nn.modules.container.Sequential",
• "torch.distributions.normal.Normal",
• "torch.nn.modules.container.ParameterList",
• "transformers.generation.configuration_utils.GenerationConfig",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LlamaMoEForCausalLM",
• "torch.nn.modules.activation.Softmax",
• "quanto.nn.qlinear.QLinear",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.UniversalCalculator",
• "collections.OrderedDict",
• "torch.BFloat16Storage",
• "torch.Size",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.modeling_llama_moe_hf.LlamaAttention",
• "__builtin__.set",
• "torch.FloatStorage",
• "transformers_modules.llama-moe.LLaMA-MoE-v1-3_5B-2_8.177a9a71ac1799afb91a5d11b8672ec26945620d.configuration_llama_moe.LlamaMoEConfig"

How to fix it?