Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
997df46a
Unverified
Commit
997df46a
authored
Jul 10, 2024
by
Woosuk Kwon
Committed by
GitHub
Jul 10, 2024
Browse files
[Bugfix][Neuron] Fix soft prompt method error in NeuronExecutor (#6313)
parent
ae151d73
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
0 deletions
+16
-0
vllm/executor/neuron_executor.py
vllm/executor/neuron_executor.py
+16
-0
No files found.
vllm/executor/neuron_executor.py
View file @
997df46a
...
@@ -70,6 +70,22 @@ class NeuronExecutor(ExecutorBase):
...
@@ -70,6 +70,22 @@ class NeuronExecutor(ExecutorBase):
def
list_loras
(
self
)
->
Set
[
int
]:
def
list_loras
(
self
)
->
Set
[
int
]:
return
self
.
driver_worker
.
list_loras
()
return
self
.
driver_worker
.
list_loras
()
def
add_prompt_adapter
(
self
,
prompt_adapter_request
)
->
bool
:
raise
NotImplementedError
(
"Soft prompt is currently not supported by the Neuron backend."
)
def
remove_prompt_adapter
(
self
,
prompt_adapter_id
:
int
)
->
bool
:
raise
NotImplementedError
(
"Soft prompt is currently not supported by the Neuron backend."
)
def
pin_prompt_adapter
(
self
,
prompt_adapter_id
:
int
)
->
bool
:
raise
NotImplementedError
(
"Soft prompt is currently not supported by the Neuron backend."
)
def
list_prompt_adapters
(
self
)
->
Set
[
int
]:
raise
NotImplementedError
(
"Soft prompt is currently not supported by the Neuron backend."
)
def
check_health
(
self
)
->
None
:
def
check_health
(
self
)
->
None
:
# NeuronExecutor will always be healthy as long as
# NeuronExecutor will always be healthy as long as
# it's running.
# it's running.
...
...
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