Spaces:
Sleeping
Sleeping
removed torch.compile
Browse files- detector.py +0 -6
detector.py
CHANGED
|
@@ -62,12 +62,6 @@ del ckpt
|
|
| 62 |
encoder = encoder.to(DEVICE).to(torch.bfloat16)
|
| 63 |
encoder.eval()
|
| 64 |
|
| 65 |
-
# Optional: PyTorch 2.0 Compilation for faster execution
|
| 66 |
-
try:
|
| 67 |
-
encoder = torch.compile(encoder)
|
| 68 |
-
except Exception as e:
|
| 69 |
-
print(f"torch.compile skipped: {e}")
|
| 70 |
-
|
| 71 |
print("Encoder ready.")
|
| 72 |
|
| 73 |
print("Loading AttentiveClassifier probe...")
|
|
|
|
| 62 |
encoder = encoder.to(DEVICE).to(torch.bfloat16)
|
| 63 |
encoder.eval()
|
| 64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
print("Encoder ready.")
|
| 66 |
|
| 67 |
print("Loading AttentiveClassifier probe...")
|