Upload app.py

app.py

@@ -46,8 +46,8 @@ parser.add_argument('--if_control', type=bool, default=True)
 args = parser.parse_args()
 
 
-
-args.device = "cuda"
+args.device = "cuda" if torch.cuda.is_available() else "cpu"
+dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
 
 vae = AutoencoderKL.from_pretrained('stabilityai/sd-vae-ft-mse').to(dtype=torch.float16, device=args.device)
 tokenizer = CLIPTokenizer.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", subfolder="tokenizer")