Hugging Face's logo

Tu2003716
/
COCOM_disabled_flash_attn

Safetensors
COCOM
custom_code
Model card Files Community
8

Remove Longformer

#7
by Plasmarine - opened
base: refs/heads/main
←
from: refs/pr/7
Discussion Files changed
+1
-1
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