Upload app.py

app.py

@@ -47,7 +47,7 @@ args = parser.parse_args()
 
 
 args.device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
+dtype = torch.float16 if str(args.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")