Skip to content

Commit

Permalink
[Bugfix] Use RayActorError for older versions of Ray in RayTokenizerG…
Browse files Browse the repository at this point in the history
…roupPool (vllm-project#6039)

Signed-off-by: Alvant <alvasian@yandex.ru>
  • Loading branch information
Yard1 authored and Alvant committed Oct 26, 2024
1 parent f5b0768 commit 5152279
Showing 1 changed file with 5 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,11 @@
import os
from typing import List, Optional

from ray.exceptions import ActorDiedError
try:
from ray.exceptions import ActorDiedError
except ImportError:
# For older versions of Ray
from ray.exceptions import RayActorError as ActorDiedError
from ray.util.scheduling_strategies import NodeAffinitySchedulingStrategy
from transformers import PreTrainedTokenizer

Expand Down

0 comments on commit 5152279

Please sign in to comment.