From ee1ee05b263d3717a6c39111587de38bebee852e Mon Sep 17 00:00:00 2001 From: ZeroCool940711 Date: Sat, 3 Dec 2022 06:04:04 -0700 Subject: [PATCH] Added option to the Settings page to enable or disable cudnn. --- scripts/Settings.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/scripts/Settings.py b/scripts/Settings.py index cf1de5d..28a94db 100644 --- a/scripts/Settings.py +++ b/scripts/Settings.py @@ -125,8 +125,13 @@ def layout(): st.session_state["defaults"].general.no_half = st.checkbox("No Half", value=st.session_state['defaults'].general.no_half, help="DO NOT switch the model to 16-bit floats. Default: False") + st.session_state["defaults"].general.use_cudnn = st.checkbox("Use cudnn", value=st.session_state['defaults'].general.use_cudnn, + help="Switch the pytorch backend to use cudnn, this should help with fixing Nvidia 16xx cards getting" + "a black or green image. Default: False") + st.session_state["defaults"].general.use_float16 = st.checkbox("Use float16", value=st.session_state['defaults'].general.use_float16, help="Switch the model to 16-bit floats. Default: False") + precision_list = ['full', 'autocast'] st.session_state["defaults"].general.precision = st.selectbox("Precision", precision_list, index=precision_list.index(st.session_state['defaults'].general.precision),