Skip to content

GitLab

Commit 19568c52 authored by Casper Hansen's avatar Casper Hansen
Browse files

Remove comment

parent 984fd2f8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 8 deletions
awq/models/gpt_bigcode.py
View file @ 19568c52
...@@ -35,14 +35,7 @@ class GptBigCodeAWQForCausalLM(BaseAWQForCausalLM): ...@@ -35,14 +35,7 @@ class GptBigCodeAWQForCausalLM(BaseAWQForCausalLM):
module2inspect=module.attn, module2inspect=module.attn,
kwargs=module_kwargs kwargs=module_kwargs
)) ))
# attention output
# layers.append(dict(
# prev_op=module.attn.c_attn,
# layers=[module.attn.c_proj],
# inp=input_feat['attn.c_proj']
# ))
# linear 1 # linear 1
layers.append(dict( layers.append(dict(
prev_op=module.ln_2, prev_op=module.ln_2,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment