layerdiffusion commited on
Commit
7e6fbbc
·
1 Parent(s): 91f1528
Files changed (1) hide show
  1. demo_gradio.py +1 -1
demo_gradio.py CHANGED
@@ -37,7 +37,7 @@ args = parser.parse_args()
37
  print(args)
38
 
39
  free_mem_gb = get_cuda_free_memory_gb(gpu)
40
- high_vram = free_mem_gb > 40
41
 
42
  print(f'Free VRAM {free_mem_gb} GB')
43
  print(f'High-VRAM Mode: {high_vram}')
 
37
  print(args)
38
 
39
  free_mem_gb = get_cuda_free_memory_gb(gpu)
40
+ high_vram = free_mem_gb > 60
41
 
42
  print(f'Free VRAM {free_mem_gb} GB')
43
  print(f'High-VRAM Mode: {high_vram}')