free_mem_gb = 0
Browse files
app.py
CHANGED
|
@@ -46,6 +46,7 @@ from diffusers_helper.bucket_tools import find_nearest_bucket
|
|
| 46 |
from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig, HunyuanVideoTransformer3DModel, HunyuanVideoPipeline
|
| 47 |
|
| 48 |
high_vram = False
|
|
|
|
| 49 |
|
| 50 |
if torch.cuda.device_count() > 0:
|
| 51 |
free_mem_gb = get_cuda_free_memory_gb(gpu)
|
|
|
|
| 46 |
from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig, HunyuanVideoTransformer3DModel, HunyuanVideoPipeline
|
| 47 |
|
| 48 |
high_vram = False
|
| 49 |
+
free_mem_gb = 0
|
| 50 |
|
| 51 |
if torch.cuda.device_count() > 0:
|
| 52 |
free_mem_gb = get_cuda_free_memory_gb(gpu)
|