Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ root.addHandler(handler2)
|
|
47 |
# constant data
|
48 |
|
49 |
dtype = torch.float16
|
50 |
-
device = torch.cuda.is_available()
|
51 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
52 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
|
53 |
|
|
|
47 |
# constant data
|
48 |
|
49 |
dtype = torch.float16
|
50 |
+
device = torch.cuda.is_available() if "cuda" else "cpu"
|
51 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
52 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
|
53 |
|