Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ import jax
|
|
| 23 |
import jax.numpy as jnp
|
| 24 |
|
| 25 |
last_motion=None
|
| 26 |
-
fps=
|
| 27 |
time=5
|
| 28 |
device = "cuda"
|
| 29 |
dtype = torch.float16
|
|
@@ -139,6 +139,8 @@ def Piper(image,positive,negative,motion):
|
|
| 139 |
pipe.load_lora_weights(motion, adapter_name="motion")
|
| 140 |
pipe.set_adapters(["motion"], [0.7])
|
| 141 |
last_motion = motion
|
|
|
|
|
|
|
| 142 |
|
| 143 |
return pipe(
|
| 144 |
positive,
|
|
|
|
| 23 |
import jax.numpy as jnp
|
| 24 |
|
| 25 |
last_motion=None
|
| 26 |
+
fps=15
|
| 27 |
time=5
|
| 28 |
device = "cuda"
|
| 29 |
dtype = torch.float16
|
|
|
|
| 139 |
pipe.load_lora_weights(motion, adapter_name="motion")
|
| 140 |
pipe.set_adapters(["motion"], [0.7])
|
| 141 |
last_motion = motion
|
| 142 |
+
|
| 143 |
+
pipe.to(device)
|
| 144 |
|
| 145 |
return pipe(
|
| 146 |
positive,
|