Spaces:

charliebaby2023
/

Fast_Stable_diffusion_CPU

Paused

charliebaby2023 commited on Feb 25

Commit

f69ea65

verified ·

1 Parent(s): aae0a97

Update app_demo.py

Files changed (1) hide show

app_demo.py CHANGED Viewed

@@ -37,7 +37,8 @@ pipe = DiffusionPipeline.from_pretrained(  "charliebaby2023/p_y"  ,    custom_pi
     pipe.safety_checker = None,
  use_safetensors=True
 )
-pipe.to(torch_device="cpu",torch_dtype="float16",  torch_dtype=DTYPE)
 pipe.to("cpu")

     pipe.safety_checker = None,
  use_safetensors=True
 )
+#pipe.to(torch_device="cpu",torch_dtype="float16",  torch_dtype=DTYPE)
+pipe.to(torch_dtype="float16" )
 pipe.to("cpu")