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
4cfed9e7
Commit
4cfed9e7
authored
Nov 28, 2024
by
yangql
Browse files
Update dateset/alpaca_data_cleaned.json
Deleted offline_inference.py
parent
fb40e100
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
0 additions
and
23 deletions
+0
-23
dateset/alpaca_data_cleaned.json
dateset/alpaca_data_cleaned.json
+0
-0
offline_inference.py
offline_inference.py
+0
-23
No files found.
examples/quantization/
dat
a
set/alpaca_data_cleaned.json
→
dat
e
set/alpaca_data_cleaned.json
View file @
4cfed9e7
File moved
offline_inference.py
deleted
100644 → 0
View file @
fb40e100
from
vllm
import
LLM
,
SamplingParams
if
__name__
==
'__main__'
:
# Sample prompts.
prompts
=
[
"Hello, my name is"
,
"The president of the United States is"
,
"The capital of France is"
,
"The future of AI is"
,
]
# Create a sampling params object.
sampling_params
=
SamplingParams
(
temperature
=
0.8
,
top_p
=
0.95
,
max_tokens
=
16
)
# Create an LLM.
llm
=
LLM
(
model
=
"./Qwen1.5-7B-4bit-gptq-2"
,
tensor_parallel_size
=
1
,
dtype
=
"float16"
,
trust_remote_code
=
True
,
enforce_eager
=
True
)
# Generate texts from the prompts. The output is a list of RequestOutput objects
# that contain the prompt, generated text, and other information.
outputs
=
llm
.
generate
(
prompts
,
sampling_params
)
# Print the outputs.
for
output
in
outputs
:
prompt
=
output
.
prompt
generated_text
=
output
.
outputs
[
0
].
text
print
(
f
"Prompt:
{
prompt
!
r
}
, Generated text:
{
generated_text
!
r
}
"
)
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