vaibhavad commited on
Commit
31474e3
·
verified ·
1 Parent(s): 32e5787

Update modeling_llama_encoder.py

Browse files
Files changed (1) hide show
  1. modeling_llama_encoder.py +1 -1
modeling_llama_encoder.py CHANGED
@@ -46,7 +46,7 @@ class ModifiedLlamaDecoderLayer(LlamaDecoderLayer):
46
  )
47
 
48
 
49
- class LlamaBiModel(LlamaModel):
50
  def __init__(self, config):
51
  if not is_transformers_attn_greater_or_equal_4_56_2():
52
  raise ValueError(
 
46
  )
47
 
48
 
49
+ class LlamaEncoderModel(LlamaModel):
50
  def __init__(self, config):
51
  if not is_transformers_attn_greater_or_equal_4_56_2():
52
  raise ValueError(