Bo-Kyeong Kim
commited on
Commit
•
3fa2188
1
Parent(s):
3b82b20
change device
Browse files
demo.py
CHANGED
|
@@ -7,8 +7,8 @@ import time
|
|
| 7 |
ORIGINAL_CHECKPOINT_ID = "CompVis/stable-diffusion-v1-4"
|
| 8 |
COMPRESSED_UNET_PATH = "checkpoints/BK-SDM-Small_iter50000"
|
| 9 |
|
| 10 |
-
|
| 11 |
-
DEVICE='cpu'
|
| 12 |
|
| 13 |
class SdmCompressionDemo:
|
| 14 |
def __init__(self, device) -> None:
|
|
|
|
| 7 |
ORIGINAL_CHECKPOINT_ID = "CompVis/stable-diffusion-v1-4"
|
| 8 |
COMPRESSED_UNET_PATH = "checkpoints/BK-SDM-Small_iter50000"
|
| 9 |
|
| 10 |
+
DEVICE='cuda'
|
| 11 |
+
# DEVICE='cpu'
|
| 12 |
|
| 13 |
class SdmCompressionDemo:
|
| 14 |
def __init__(self, device) -> None:
|