Darius Morawiec commited on
Commit
bc51dfa
·
1 Parent(s): 57edf3e

Remove `CUDA_VISIBLE_DEVICES` (for local development)

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -16,10 +16,8 @@ from transformers import (
16
  Qwen3VLForConditionalGeneration,
17
  )
18
 
19
- os.environ["CUDA_VISIBLE_DEVICES"] = "0"
20
-
21
  # Handle spaces.GPU decorator for GPU allocation in Spaces
22
- if os.environ.get("SPACES_ZERO_GPU") is not None:
23
  import spaces
24
  else:
25
 
 
16
  Qwen3VLForConditionalGeneration,
17
  )
18
 
 
 
19
  # Handle spaces.GPU decorator for GPU allocation in Spaces
20
+ if "SPACES_ZERO_GPU" in os.environ.keys():
21
  import spaces
22
  else:
23