Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
b4ac571
1
Parent(s):
3416183
suggested patching
Browse files
app.py
CHANGED
@@ -16,6 +16,8 @@ omni_zero = OmniZeroCouple(
|
|
16 |
device="cuda",
|
17 |
)
|
18 |
|
|
|
|
|
19 |
@spaces.GPU()
|
20 |
def generate(
|
21 |
base_image="https://cdn-prod.styleof.com/inferences/cm1ho5cjl14nh14jec6phg2h8/i6k59e7gpsr45ufc7l8kun0g-medium.jpeg",
|
|
|
16 |
device="cuda",
|
17 |
)
|
18 |
|
19 |
+
omni_zero.generate = spaces.GPU(omni_zero.generate)
|
20 |
+
|
21 |
@spaces.GPU()
|
22 |
def generate(
|
23 |
base_image="https://cdn-prod.styleof.com/inferences/cm1ho5cjl14nh14jec6phg2h8/i6k59e7gpsr45ufc7l8kun0g-medium.jpeg",
|