RealVis_v5.0_BF16_C2

Paused

ford442 commited on Apr 29

Commit

68f04f9

verified ·

1 Parent(s): a1c9675

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -86,7 +86,7 @@ torch.backends.cudnn.benchmark = False
 torch.backends.cuda.preferred_blas_library="cublas"
 torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
-torch.cuda.memory._record_memory_history(enabled='None', context='None', stacks='None', max_entries=9223372036854775807, device=None)
 HF_TOKEN = os.getenv("HF_TOKEN")
 FTP_HOST = os.getenv("FTP_HOST")

 torch.backends.cuda.preferred_blas_library="cublas"
 torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
+torch.cuda.memory._record_memory_history(enabled=False)
 HF_TOKEN = os.getenv("HF_TOKEN")
 FTP_HOST = os.getenv("FTP_HOST")