Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions gpu_container/vllm/reproducible_vllm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__(
Expand Down
3 changes: 2 additions & 1 deletion prompting/weight_setting/weight_setter.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -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"
Expand Down