Spaces:
Paused
Paused
Omer Karisman
commited on
Commit
•
6ddb532
1
Parent(s):
3416183
zero gpu
Browse files
app.py
CHANGED
|
@@ -9,13 +9,15 @@ import torch
|
|
| 9 |
torch.jit.script = lambda f: f
|
| 10 |
####
|
| 11 |
|
| 12 |
-
from
|
| 13 |
|
| 14 |
omni_zero = OmniZeroCouple(
|
| 15 |
base_model="frankjoshua/albedobaseXL_v13",
|
| 16 |
device="cuda",
|
| 17 |
)
|
| 18 |
|
|
|
|
|
|
|
| 19 |
@spaces.GPU()
|
| 20 |
def generate(
|
| 21 |
base_image="https://cdn-prod.styleof.com/inferences/cm1ho5cjl14nh14jec6phg2h8/i6k59e7gpsr45ufc7l8kun0g-medium.jpeg",
|
|
|
|
| 9 |
torch.jit.script = lambda f: f
|
| 10 |
####
|
| 11 |
|
| 12 |
+
from omni_zero_spaces import OmniZeroCouple
|
| 13 |
|
| 14 |
omni_zero = OmniZeroCouple(
|
| 15 |
base_model="frankjoshua/albedobaseXL_v13",
|
| 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",
|