Spaces:
Running
on
Zero
Running
on
Zero
Update utils/common_viz.py
Browse files- utils/common_viz.py +0 -2
utils/common_viz.py
CHANGED
|
@@ -4,7 +4,6 @@ import clip
|
|
| 4 |
from hydra import compose, initialize
|
| 5 |
from hydra.utils import instantiate
|
| 6 |
from omegaconf import OmegaConf
|
| 7 |
-
import spaces
|
| 8 |
import torch
|
| 9 |
from torchtyping import TensorType
|
| 10 |
from torch.utils.data import DataLoader
|
|
@@ -105,7 +104,6 @@ def get_batch(
|
|
| 105 |
return batch
|
| 106 |
|
| 107 |
|
| 108 |
-
@spaces.GPU
|
| 109 |
def init(
|
| 110 |
config_name: str,
|
| 111 |
) -> Tuple[Diffuser, clip.model.CLIP, MultimodalDataset, torch.device]:
|
|
|
|
| 4 |
from hydra import compose, initialize
|
| 5 |
from hydra.utils import instantiate
|
| 6 |
from omegaconf import OmegaConf
|
|
|
|
| 7 |
import torch
|
| 8 |
from torchtyping import TensorType
|
| 9 |
from torch.utils.data import DataLoader
|
|
|
|
| 104 |
return batch
|
| 105 |
|
| 106 |
|
|
|
|
| 107 |
def init(
|
| 108 |
config_name: str,
|
| 109 |
) -> Tuple[Diffuser, clip.model.CLIP, MultimodalDataset, torch.device]:
|