update base image
This commit is contained in:
@@ -8,9 +8,6 @@ import vllm.envs as envs
|
||||
from vllm.logger import init_logger
|
||||
|
||||
|
||||
# fix bfloat16 double size issue
|
||||
torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
|
||||
|
||||
logger = init_logger(__name__)
|
||||
|
||||
class KunlunPlatform(Platform):
|
||||
|
||||
Reference in New Issue
Block a user