Spaces:

wanghuging
/

demo-for-skin

Sleeping

wanghuging commited on Dec 11, 2023

Commit

8f57828

1 Parent(s): 83db897

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ device = torch.device(
     "cuda" if torch.cuda.is_available() else "xpu" if xpu_available else "cpu"
 )
 torch_device = device
-torch_dtype = torch.float16
 print(f"SAFETY_CHECKER: {SAFETY_CHECKER}")
 print(f"TORCH_COMPILE: {TORCH_COMPILE}")

     "cuda" if torch.cuda.is_available() else "xpu" if xpu_available else "cpu"
 )
 torch_device = device
+torch_dtype = torch.float32 #float16
 print(f"SAFETY_CHECKER: {SAFETY_CHECKER}")
 print(f"TORCH_COMPILE: {TORCH_COMPILE}")