Hugging Face's logo

Spaces:
jingyangcarl
/
matgen
Sleeping

App Files Community
jingyangcarl commited on
Commit
fd665d1
·
1 Parent(s): a7740b0

debug zerogpu

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -65,7 +65,7 @@ class Model:
65
  pipe.to(self.device)
66
  if self.device.type == "cuda":
67
  import os
68
- if os.environ.get("SPACES_ZERO_GPU", "0") == "1"
69
  # when running on ZeroGPU, enable CPU offload
70
  # pipe.enable_xformers_memory_efficient_attention() doens't work
71
  # pipe.enable_model_cpu_offload()
 
65
  pipe.to(self.device)
66
  if self.device.type == "cuda":
67
  import os
68
+ if os.environ.get("SPACES_ZERO_GPU", "0") == "1":
69
  # when running on ZeroGPU, enable CPU offload
70
  # pipe.enable_xformers_memory_efficient_attention() doens't work
71
  # pipe.enable_model_cpu_offload()