Qubitium commited on
Commit
2ec0690
·
verified ·
1 Parent(s): ddfb6bf

Add files using upload-large-folder tool

Browse files
Files changed (1) hide show
  1. test_minimax_m2_hf.py +14 -14
test_minimax_m2_hf.py CHANGED
@@ -21,20 +21,20 @@ from pathlib import Path
21
  import torch.nn as nn
22
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
23
 
24
- from gptqmodel.hf_minimax_m2.modeling_minimax_m2 import (
25
- MiniMaxAttention,
26
- MiniMaxDecoderLayer,
27
- MiniMaxForCausalLM,
28
- MiniMaxMLP,
29
- MiniMaxM2Attention,
30
- MiniMaxM2DecoderLayer,
31
- MiniMaxM2ForCausalLM,
32
- MiniMaxM2MLP,
33
- MiniMaxM2RMSNorm,
34
- MiniMaxM2SparseMoeBlock,
35
- MiniMaxRMSNorm,
36
- MiniMaxSparseMoeBlock,
37
- )
38
 
39
 
40
  def parse_args() -> argparse.Namespace:
 
21
  import torch.nn as nn
22
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
23
 
24
+ # from gptqmodel.hf_minimax_m2.modeling_minimax_m2 import (
25
+ # MiniMaxAttention,
26
+ # MiniMaxDecoderLayer,
27
+ # MiniMaxForCausalLM,
28
+ # MiniMaxMLP,
29
+ # MiniMaxM2Attention,
30
+ # MiniMaxM2DecoderLayer,
31
+ # MiniMaxM2ForCausalLM,
32
+ # MiniMaxM2MLP,
33
+ # MiniMaxM2RMSNorm,
34
+ # MiniMaxM2SparseMoeBlock,
35
+ # MiniMaxRMSNorm,
36
+ # MiniMaxSparseMoeBlock,
37
+ # )
38
 
39
 
40
  def parse_args() -> argparse.Namespace: