Fabrice-TIERCELIN commited on
Commit
5c3d168
·
verified ·
1 Parent(s): 1959bdf

high_vram = False

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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