clean up a bit

This commit is contained in:
Cyberes 2023-10-20 18:26:41 -06:00
parent f39e976b34
commit e236e93a79
3 changed files with 9 additions and 10 deletions

View File

@ -222,7 +222,7 @@ class RequestHandler:
processing_ip = 0
if queued_ip_count + processing_ip >= self.token_simultaneous_ip:
print(f'Rejecting request from {self.client_ip} - {queued_ip_count} queued, {processing_ip} processing.')
print(f'Rejecting request from {self.client_ip} - {processing_ip} processing, {queued_ip_count} queued')
return True
else:
return False

View File

@ -25,6 +25,7 @@ def console_printer():
for k in processing:
processing_count += redis.get(k, default=0, dtype=int)
backends = [k for k, v in cluster_config.all().items() if v['online']]
activity = priority_queue.activity()
# Calculate the queue size the same way it's done on the stats.
queue_size = 0
@ -32,8 +33,6 @@ def console_printer():
for model in running_models:
queue_size += priority_queue.len(model)
activity = priority_queue.activity()
# Active Workers and Processing should read the same. If not, that's an issue.
logger.info(f'REQUEST QUEUE -> Active Workers: {len([i for i in activity if i[1]])} | Processing: {processing_count} | Queued: {queue_size} | Backends Online: {len(backends)}')
time.sleep(2)
time.sleep(10)

View File

@ -44,21 +44,21 @@ rows=$(echo "sqrt($num_windows)" | bc)
columns=$(echo "($num_windows + $rows - 1) / $rows" | bc)
# Create a new tmux session
tmux new-session -d -s my_session "$command -p 0"
tmux new-session -d -s llm_tester "$command -p 0"
# Create the remaining windows
for ((i = 1; i < $num_windows; i++)); do
if ((i % $columns == 0)); then
tmux select-layout -t my_session:0 tiled
tmux select-layout -t llm_tester:0 tiled
tmux select-pane -t 0
tmux split-window -t my_session:0 -v "$command -p $i"
tmux split-window -t llm_tester:0 -v "$command -p $i"
else
tmux split-window -t my_session:0 -h "$command -p $i"
tmux split-window -t llm_tester:0 -h "$command -p $i"
fi
done
# Balance the windows
tmux select-layout -t my_session:0 tiled
tmux select-layout -t llm_tester:0 tiled
# Attach to the session
tmux attach-session -t my_session
tmux attach-session -t llm_tester