Update app.py
Browse files
app.py
CHANGED
|
@@ -86,7 +86,6 @@ torch.backends.cudnn.benchmark = False
|
|
| 86 |
torch.backends.cuda.preferred_blas_library="cublas"
|
| 87 |
torch.backends.cuda.preferred_linalg_library="cusolver"
|
| 88 |
torch.set_float32_matmul_precision("highest")
|
| 89 |
-
torch.cuda.memory._record_memory_history(enabled=False)
|
| 90 |
|
| 91 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 92 |
FTP_HOST = os.getenv("FTP_HOST")
|
|
|
|
| 86 |
torch.backends.cuda.preferred_blas_library="cublas"
|
| 87 |
torch.backends.cuda.preferred_linalg_library="cusolver"
|
| 88 |
torch.set_float32_matmul_precision("highest")
|
|
|
|
| 89 |
|
| 90 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 91 |
FTP_HOST = os.getenv("FTP_HOST")
|