You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
355 B
Python
18 lines
355 B
Python
import os
|
|
|
|
import hivemind
|
|
|
|
from petals.client import *
|
|
from petals.utils.logging import initialize_logs as _initialize_logs
|
|
|
|
__version__ = "1.1.5"
|
|
|
|
|
|
def _override_bfloat16_mode_default():
|
|
if os.getenv("USE_LEGACY_BFLOAT16") is None:
|
|
hivemind.compression.base.USE_LEGACY_BFLOAT16 = False
|
|
|
|
|
|
_initialize_logs()
|
|
_override_bfloat16_mode_default()
|