regisss
/

llama2-70b-fused-qkv-mlperf

Text Generation

Model card Files Files and versions Community

regisss HF staff commited on Feb 13

Commit

ef4e9ca

•

1 Parent(s): 5aaa188

Update modeling_llama.py

Files changed (1) hide show

modeling_llama.py +1 -1

modeling_llama.py CHANGED Viewed

@@ -1270,7 +1270,7 @@ class LlamaModel(LlamaPreTrainedModel):
         return causal_mask
-class LlamaForCausalLM(LlamaPreTrainedModel):
     _tied_weights_keys = ["lm_head.weight"]
     def __init__(self, config):

         return causal_mask
+class CustomLlamaForCausalLM(LlamaPreTrainedModel):
     _tied_weights_keys = ["lm_head.weight"]
     def __init__(self, config):