From e879d1c5f3efe8f237f68ecbedc5ff65059953ea Mon Sep 17 00:00:00 2001 From: vfbd Date: Thu, 11 Aug 2022 18:42:56 -0400 Subject: [PATCH] Hide the warning about `torch.distributed.reduce_op` being deprecated --- aiserver.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/aiserver.py b/aiserver.py index 049eabc4..1532f680 100644 --- a/aiserver.py +++ b/aiserver.py @@ -37,6 +37,7 @@ import bisect import functools import traceback import inspect +import warnings from collections.abc import Iterable from typing import Any, Callable, TypeVar, Tuple, Union, Dict, Set, List, Optional, Type @@ -1958,13 +1959,15 @@ def load_model(use_gpu=True, gpu_layers=None, disk_layers=None, initial_load=Fal model = None generator = None model_config = None - for tensor in gc.get_objects(): - try: - if torch.is_tensor(tensor): - with torch.no_grad(): - tensor.set_(torch.tensor((), device=tensor.device, dtype=tensor.dtype)) - except: - pass + with torch.no_grad(): + with warnings.catch_warnings(): + warnings.filterwarnings("ignore", message="torch.distributed.reduce_op is deprecated") + for tensor in gc.get_objects(): + try: + if torch.is_tensor(tensor): + tensor.set_(torch.tensor((), device=tensor.device, dtype=tensor.dtype)) + except: + pass gc.collect() try: torch.cuda.empty_cache()