Spaces:
Running
Running
high_vram = False
Browse files
app.py
CHANGED
@@ -45,6 +45,8 @@ from diffusers_helper.clip_vision import hf_clip_vision_encode
|
|
45 |
from diffusers_helper.bucket_tools import find_nearest_bucket
|
46 |
from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig, HunyuanVideoTransformer3DModel, HunyuanVideoPipeline
|
47 |
|
|
|
|
|
48 |
if torch.cuda.device_count() > 0:
|
49 |
free_mem_gb = get_cuda_free_memory_gb(gpu)
|
50 |
high_vram = free_mem_gb > 60
|
|
|
45 |
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)
|
52 |
high_vram = free_mem_gb > 60
|