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)
  • Loading branch information
Yard1 authored and jimpang committed Jul 24, 2024
1 parent cedc094 commit f1a8359
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 f1a8359

Please sign in to comment.