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
AutoGPTQ
Commits
57512823
Commit
57512823
authored
Oct 26, 2024
by
yangql
Browse files
Update test_quant.py
parent
bf2b305e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
test_quant.py
test_quant.py
+3
-3
No files found.
test_quant.py
View file @
57512823
...
...
@@ -14,7 +14,7 @@ def gptq():
bits
=
4
,
# 4 or 8
group_size
=
128
,
damp_percent
=
0.01
,
desc_act
=
Fals
e
,
# set to False can significantly speed up inference but the perplexity may slightly bad
desc_act
=
Tru
e
,
# set to False can significantly speed up inference but the perplexity may slightly bad
static_groups
=
False
,
sym
=
True
,
true_sequential
=
True
,
...
...
@@ -46,7 +46,7 @@ def gptq():
messages
.
append
(
msg
)
print
(
'len(messages):'
,
len
(
messages
))
messages
=
messages
[:
5
]
messages
=
messages
[:
5
00
]
print
(
'len(messages):'
,
len
(
messages
))
data
=
[]
...
...
@@ -65,4 +65,4 @@ def gptq():
t2
=
time
.
time
()
print
((
'time:{:.2f}s'
).
format
(
t2
-
t1
))
gptq
()
\ No newline at end of file
gptq
()
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