Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
AutoAWQ
Commits
9204d6e8
Commit
9204d6e8
authored
Sep 20, 2023
by
Casper
Browse files
Add desc
parent
44470a7a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
awq/quantize/quantizer.py
awq/quantize/quantizer.py
+1
-1
No files found.
awq/quantize/quantizer.py
View file @
9204d6e8
...
...
@@ -52,7 +52,7 @@ class AwqQuantizer:
return
w
def
quantize
(
self
,
get_layers_for_scaling
:
function
):
for
i
in
tqdm
(
range
(
len
(
self
.
modules
)),
desc
=
""
):
for
i
in
tqdm
(
range
(
len
(
self
.
modules
)),
desc
=
"
QUANTIZING
"
):
# [STEP 1]: Get layer, extract linear modules, extract input features
self
.
modules
[
i
]
=
self
.
modules
[
i
].
cuda
()
named_linears
=
get_named_linears
(
self
.
modules
[
i
])
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment