Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
1521648
1
Parent(s):
3c4b164
Update utils/common_viz.py
Browse files- utils/common_viz.py +1 -0
utils/common_viz.py
CHANGED
|
@@ -104,6 +104,7 @@ def get_batch(
|
|
| 104 |
return batch
|
| 105 |
|
| 106 |
|
|
|
|
| 107 |
def init(
|
| 108 |
config_name: str,
|
| 109 |
) -> Tuple[Diffuser, clip.model.CLIP, MultimodalDataset, torch.device]:
|
|
|
|
| 104 |
return batch
|
| 105 |
|
| 106 |
|
| 107 |
+
@spaces.GPU
|
| 108 |
def init(
|
| 109 |
config_name: str,
|
| 110 |
) -> Tuple[Diffuser, clip.model.CLIP, MultimodalDataset, torch.device]:
|