Plasmarine commited on
Commit
196e5f1
·
verified ·
1 Parent(s): 8546634

Remove Longformer

Browse files
Files changed (1) hide show
  1. modeling_cocom.py +1 -1
modeling_cocom.py CHANGED
@@ -1,4 +1,4 @@
1
- from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, PreTrainedModel, PretrainedConfig, AutoModel, LongformerForCausalLM, LongformerTokenizer
2
  from linformer.attention import LinformerSelfAttention
3
  import torch
4
  import math
 
1
+ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, PreTrainedModel, PretrainedConfig, AutoModel
2
  from linformer.attention import LinformerSelfAttention
3
  import torch
4
  import math