Spaces:
Running
on
Zero
Running
on
Zero
more on gpu...
Browse files
model.py
CHANGED
|
@@ -21,6 +21,7 @@ class Model:
|
|
| 21 |
def __init__(self):
|
| 22 |
self.max_num_faces = 100000
|
| 23 |
|
|
|
|
| 24 |
def load_config(self, shape_path: str, text: str, seed: int,
|
| 25 |
guidance_scale: float) -> TrainConfig:
|
| 26 |
text += ', {} view'
|
|
|
|
| 21 |
def __init__(self):
|
| 22 |
self.max_num_faces = 100000
|
| 23 |
|
| 24 |
+
@spaces.GPU
|
| 25 |
def load_config(self, shape_path: str, text: str, seed: int,
|
| 26 |
guidance_scale: float) -> TrainConfig:
|
| 27 |
text += ', {} view'
|