Spaces:

S-4-G-4-R
/

NOVA

Running on Zero

S-4-G-4-R commited on 6 days ago

Commit

4246a14

verified ·

1 Parent(s): 8721f9c

Update ui/gpu.py

Files changed (1) hide show

ui/gpu.py CHANGED Viewed

@@ -28,7 +28,7 @@ import spaces
 ON_ZEROGPU = os.getenv("SPACES_ZERO_GPU", "").lower() in ("1", "t", "true")
 # The device to use *inside* a GPU window. Outside one, always "cpu".
-GPU_DEVICE = cpu  #"cuda" if ON_ZEROGPU else "cpu"
 # Generous but bounded. The window has to cover PDF text extraction and chunking
 # (CPU work that unavoidably happens inside build_vectorstore) plus the encode

 ON_ZEROGPU = os.getenv("SPACES_ZERO_GPU", "").lower() in ("1", "t", "true")
 # The device to use *inside* a GPU window. Outside one, always "cpu".
+GPU_DEVICE = "cpu"  #"cuda" if ON_ZEROGPU else "cpu"
 # Generous but bounded. The window has to cover PDF text extraction and chunking
 # (CPU work that unavoidably happens inside build_vectorstore) plus the encode