PrunaAI

/

llama-moe-LLaMA-MoE-v1-3_0B-2_16-QUANTO-float8bit-smashed

like 0

Follow

Pruna AI 146

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

How to fix it?