Spaces:
Runtime error
Runtime error
ZeroGPU Test 2
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ def get_device():
|
|
16 |
else:
|
17 |
return "cpu"
|
18 |
|
|
|
19 |
def make_uncond_seq(seq_len, model_type):
|
20 |
if model_type == "EvoDiff-Seq-OADM 38M":
|
21 |
checkpoint = OA_DM_38M()
|
|
|
16 |
else:
|
17 |
return "cpu"
|
18 |
|
19 |
+
@spaces.GPU
|
20 |
def make_uncond_seq(seq_len, model_type):
|
21 |
if model_type == "EvoDiff-Seq-OADM 38M":
|
22 |
checkpoint = OA_DM_38M()
|