80 -> 60
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig, Hunyuan
|
|
| 46 |
|
| 47 |
if torch.cuda.device_count() > 0:
|
| 48 |
free_mem_gb = get_cuda_free_memory_gb(gpu)
|
| 49 |
-
high_vram = free_mem_gb >
|
| 50 |
|
| 51 |
print(f'Free VRAM {free_mem_gb} GB')
|
| 52 |
print(f'High-VRAM Mode: {high_vram}')
|
|
|
|
| 46 |
|
| 47 |
if torch.cuda.device_count() > 0:
|
| 48 |
free_mem_gb = get_cuda_free_memory_gb(gpu)
|
| 49 |
+
high_vram = free_mem_gb > 60
|
| 50 |
|
| 51 |
print(f'Free VRAM {free_mem_gb} GB')
|
| 52 |
print(f'High-VRAM Mode: {high_vram}')
|