[Bugfix] Use RayActorError for older versions of Ray in RayTokenizerGroupPool (#6039)

This commit is contained in:
Antoni Baum
2024-07-01 13:12:40 -07:00
committed by GitHub
parent 8893130b63
commit dec6fc6f3b

View File

@ -2,7 +2,11 @@ import asyncio
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