Update app.py

app.py

@@ -23,7 +23,8 @@ if is_colab:
 else:
     import streamlit as st
     scheduler = DDIMScheduler.from_config(model_id_or_path, use_auth_token=st.secrets["USER_TOKEN"], subfolder="scheduler")
-    pipe = CycleDiffusionPipeline.from_pretrained(model_id_or_path, use_auth_token=st.secrets["USER_TOKEN"], scheduler=scheduler)
+    torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+    pipe = CycleDiffusionPipeline.from_pretrained(model_id_or_path, use_auth_token=st.secrets["USER_TOKEN"], scheduler=scheduler, torch_dtype=torch_dtype)
 tokenizer = pipe.tokenizer
 
 if torch.cuda.is_available():