Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -12,10 +12,10 @@ from .models import DalleBartEncoder, DalleBartDecoder, VQGanDetokenizer
 import streamlit as st
 import time
 
-torch.set_grad_enabled(False)
-torch.set_num_threads(os.cpu_count())
-torch.backends.cudnn.enabled = False
-torch.backends.cudnn.allow_tf32 = False
+torch.set_grad_enabled(True)
+torch.set_num_threads(1)
+torch.backends.cudnn.enabled = True
+torch.backends.cudnn.allow_tf32 = True
 
 MIN_DALLE_REPO = 'https://huggingface.co/kuprel/min-dalle/resolve/main/'
 IMAGE_TOKEN_COUNT = 256