Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 3, 2022

Commit

5c25726

•

1 Parent(s): 9ca70de

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

@@ -13,7 +13,7 @@ import streamlit as st
 import time
 torch.set_grad_enabled(False)
-torch.set_num_threads(torch.set_num_threads(os.cpu_count()))
 torch.backends.cudnn.enabled = True
 torch.backends.cudnn.allow_tf32 = True

 import time
 torch.set_grad_enabled(False)
+torch.set_num_threads(os.cpu_count())
 torch.backends.cudnn.enabled = True
 torch.backends.cudnn.allow_tf32 = True