Archives/petals
2
0
Fork 0
mirror of https://github.com/bigscience-workshop/petals synced 2024-11-16 06:12:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

use logger

Browse Source
This commit is contained in:
justheuristic 2022-06-14 08:25:06 +03:00
parent ce5dedd2c7
commit 3215945882
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cli/quantize_cpu_naive.py
View File

@ -5,9 +5,10 @@ import os
import psutil
import torch.backends.quantized
import transformers
from hivemind.utils.logging import get_logger
from hivemind.utils.logging import get_logger, use_hivemind_log_handler
from tqdm.auto import trange
use_hivemind_log_handler("in_root_logger")
logger = get_logger(__file__)
DTYPE_MAP = dict(bfloat16=torch.bfloat16, float16=torch.float16, float32=torch.float32, auto="auto")