Spaces:
Running on Zero
Running on Zero
| """One lock for all GPU work of this process (local ESMFold, ESM-2, ProtT5, the checkpoints). | |
| Requests are served from a thread pool; serialising GPU work keeps memory bounded on one GPU and lets local | |
| ESMFold switch TF32 on for its own forward pass without affecting the other models. | |
| """ | |
| import threading | |
| GPU_LOCK = threading.RLock() | |