Yaron Koresh commited on
Commit
1eb986b
·
verified ·
1 Parent(s): 2c20848

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -23,7 +23,7 @@ import jax
23
  import jax.numpy as jnp
24
 
25
  last_motion=None
26
- fps=25
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,