diff --git a/gpu_container/vllm/reproducible_vllm.py b/gpu_container/vllm/reproducible_vllm.py index dd2f532e9..c659e4651 100644 --- a/gpu_container/vllm/reproducible_vllm.py +++ b/gpu_container/vllm/reproducible_vllm.py @@ -3,8 +3,11 @@ import numpy as np import torch +from tqdm import tqdm from vllm import LLM, SamplingParams +tqdm.disable = True + class ReproducibleVLLM: def __init__( diff --git a/prompting/weight_setting/weight_setter.py b/prompting/weight_setting/weight_setter.py index 6cebbc639..44064dca3 100644 --- a/prompting/weight_setting/weight_setter.py +++ b/prompting/weight_setting/weight_setter.py @@ -100,7 +100,8 @@ async def set_weights( uids=processed_weight_uids, weights=processed_weights ) except Exception as ex: - logger.exception(f"Issue with setting weights: {ex}") + logger.exception(f"Skipping weight setting: {ex}") + return # Create a dataframe from weights and uids and save it as a csv file, with the current step as the filename. if shared_settings.LOG_WEIGHTS: diff --git a/pyproject.toml b/pyproject.toml index ebad453d1..69ef99829 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "prompting" -version = "2.19.7" +version = "2.19.8" description = "Subnetwork 1 runs on Bittensor and is maintained by Macrocosmos. It's an effort to create decentralised AI" authors = ["Kalei Brady, Dmytro Bobrenko, Felix Quinque, Steffen Cruz, Richard Wardle"] readme = "README.md"