[Bugfix] Use RayActorError for older versions of Ray in RayTokenizerGroupPool (#6039)
This commit is contained in:
parent
8893130b63
commit
dec6fc6f3b
@ -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
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user