diff --git a/demo_gradio.py b/demo_gradio.py index b150c59..632cb87 100644 --- a/demo_gradio.py +++ b/demo_gradio.py @@ -37,7 +37,7 @@ args = parser.parse_args() print(args) free_mem_gb = get_cuda_free_memory_gb(gpu) -high_vram = free_mem_gb > 40 +high_vram = free_mem_gb > 60 print(f'Free VRAM {free_mem_gb} GB') print(f'High-VRAM Mode: {high_vram}')