Removed the suppress_st_warning argument from st.experimental_memo as it was causing some issues on the latest version. (#1788)

This commit is contained in:
Alejandro Gil 2023-06-11 23:19:09 -07:00 committed by GitHub
commit 2ca54a6934
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1325,7 +1325,7 @@ def torch_gc():
torch.cuda.ipc_collect()
@retry(tries=5)
#@st.experimental_memo(persist="disk", show_spinner=False, suppress_st_warning=True)
#@st.experimental_memo(persist="disk", show_spinner=False)
def load_GFPGAN(model_name='GFPGANv1.4'):
#model_name = 'GFPGANv1.3'
@ -1757,7 +1757,7 @@ def slerp(device, t, v0:torch.Tensor, v1:torch.Tensor, DOT_THRESHOLD=0.9995):
return v2
#
@st.experimental_memo(persist="disk", show_spinner=False, suppress_st_warning=True)
@st.experimental_memo(persist="disk", show_spinner=False)
def optimize_update_preview_frequency(current_chunk_speed, previous_chunk_speed_list, update_preview_frequency, update_preview_frequency_list):
"""Find the optimal update_preview_frequency value maximizing
performance while minimizing the time between updates."""