Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -42,15 +42,15 @@ def generate(slider_x, slider_y, prompt, iterations, steps,
|
|
| 42 |
|
| 43 |
@spaces.GPU
|
| 44 |
def update_x(x,y,prompt, steps, avg_diff_x_1, avg_diff_x_2, avg_diff_y_1, avg_diff_y_2):
|
| 45 |
-
avg_diff =
|
| 46 |
-
avg_diff_2nd =
|
| 47 |
image = clip_slider.generate(prompt, scale=x, scale_2nd=y, num_inference_steps=steps, avg_diff=avg_diff,avg_diff_2nd=avg_diff_2nd)
|
| 48 |
return image
|
| 49 |
|
| 50 |
@spaces.GPU
|
| 51 |
def update_y(x,y,prompt, steps, avg_diff_x_1, avg_diff_x_2, avg_diff_y_1, avg_diff_y_2):
|
| 52 |
-
avg_diff =
|
| 53 |
-
avg_diff_2nd =
|
| 54 |
image = clip_slider.generate(prompt, scale=x, scale_2nd=y, num_inference_steps=steps, avg_diff=avg_diff,avg_diff_2nd=avg_diff_2nd)
|
| 55 |
return image
|
| 56 |
|
|
|
|
| 42 |
|
| 43 |
@spaces.GPU
|
| 44 |
def update_x(x,y,prompt, steps, avg_diff_x_1, avg_diff_x_2, avg_diff_y_1, avg_diff_y_2):
|
| 45 |
+
avg_diff = (avg_diff_x_1.cuda(), avg_diff_x_2.cuda())
|
| 46 |
+
avg_diff_2nd = (avg_diff_y_1.cuda(), avg_diff_y_2.cuda())
|
| 47 |
image = clip_slider.generate(prompt, scale=x, scale_2nd=y, num_inference_steps=steps, avg_diff=avg_diff,avg_diff_2nd=avg_diff_2nd)
|
| 48 |
return image
|
| 49 |
|
| 50 |
@spaces.GPU
|
| 51 |
def update_y(x,y,prompt, steps, avg_diff_x_1, avg_diff_x_2, avg_diff_y_1, avg_diff_y_2):
|
| 52 |
+
avg_diff = (avg_diff_x_1.cuda(), avg_diff_x_2.cuda())
|
| 53 |
+
avg_diff_2nd = (avg_diff_y_1.cuda(), avg_diff_y_2.cuda())
|
| 54 |
image = clip_slider.generate(prompt, scale=x, scale_2nd=y, num_inference_steps=steps, avg_diff=avg_diff,avg_diff_2nd=avg_diff_2nd)
|
| 55 |
return image
|
| 56 |
|