Update app_demo.py
Browse files- app_demo.py +2 -1
app_demo.py
CHANGED
|
@@ -37,7 +37,8 @@ pipe = DiffusionPipeline.from_pretrained( "charliebaby2023/p_y" , custom_pi
|
|
| 37 |
pipe.safety_checker = None,
|
| 38 |
use_safetensors=True
|
| 39 |
)
|
| 40 |
-
pipe.to(torch_device="cpu",torch_dtype="float16", torch_dtype=DTYPE)
|
|
|
|
| 41 |
pipe.to("cpu")
|
| 42 |
|
| 43 |
|
|
|
|
| 37 |
pipe.safety_checker = None,
|
| 38 |
use_safetensors=True
|
| 39 |
)
|
| 40 |
+
#pipe.to(torch_device="cpu",torch_dtype="float16", torch_dtype=DTYPE)
|
| 41 |
+
pipe.to(torch_dtype="float16" )
|
| 42 |
pipe.to("cpu")
|
| 43 |
|
| 44 |
|