Update gradio_app.py

gradio_app.py

@@ -155,7 +155,7 @@ def get_image(image, sketch, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, see
 
     # text cond
     transposed = False 
-    with torch.no_grad(), torch.cuda.amp.autocast(torch.float16):
+    with torch.no_grad(), torch.cuda.amp.autocast(dtype=torch.float16):
         text_emb = model.get_learned_conditioning([prompt])
         print("before control")
         #control cond