From 92b31599090c74cee895b3211562ff16fb6ca977 Mon Sep 17 00:00:00 2001 From: hlky <106811348+hlky@users.noreply.github.com> Date: Tue, 13 Sep 2022 13:26:36 +0100 Subject: [PATCH] half precision streamlit txt2vid `RuntimeError: expected scalar type Half but found Float` with both `torch_dtype=torch.float16` and `revision="fp16"` --- scripts/webui_streamlit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/webui_streamlit.py b/scripts/webui_streamlit.py index 341fa2a..6d19d21 100644 --- a/scripts/webui_streamlit.py +++ b/scripts/webui_streamlit.py @@ -1951,7 +1951,7 @@ def txt2vid( weights_path, use_local_file=True, use_auth_token=True, - torch_dtype=torch.float16 if not defaults.general.no_half else None, + #torch_dtype=torch.float16 if not defaults.general.no_half else None, revision="fp16" if not defaults.general.no_half else None ) @@ -1971,7 +1971,7 @@ def txt2vid( weights_path, use_local_file=True, use_auth_token=True, - torch_dtype=torch.float16 if not defaults.general.no_half else None, + #torch_dtype=torch.float16 if not defaults.general.no_half else None, revision="fp16" if not defaults.general.no_half else None )