From 5d1d0bd11fc4ec21d2d28d15ac50e922b3f7960c Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 12 Apr 2023 18:24:19 -0300 Subject: [PATCH] Add the GPU index to the label --- server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server.py b/server.py index e24f2078..896b2097 100644 --- a/server.py +++ b/server.py @@ -301,7 +301,7 @@ def create_model_menus(): with gr.Row(): with gr.Column(): for i in range(len(total_mem)): - components[f'gpu_memory_{i}'] = gr.Slider(label="gpu-memory in MiB", maximum=total_mem[i], value=default_gpu_mem[i]) + components[f'gpu_memory_{i}'] = gr.Slider(label=f"gpu-memory in MiB for device :{i}", maximum=total_mem[i], value=default_gpu_mem[i]) components['cpu_memory'] = gr.Slider(label="cpu-memory in MiB", maximum=total_cpu_mem, value=default_cpu_mem) with gr.Column():