Spaces:
Running
on
Zero
Running
on
Zero
Update ace_inference.py
Browse files- ace_inference.py +1 -1
ace_inference.py
CHANGED
@@ -11,7 +11,7 @@ import torch.nn.functional as F
|
|
11 |
import torchvision.transforms.functional as TF
|
12 |
from PIL import Image
|
13 |
import torchvision.transforms as T
|
14 |
-
from scepter.modules.model.registry import DIFFUSIONS
|
15 |
from scepter.modules.model.utils.basic_utils import check_list_of_list
|
16 |
from scepter.modules.model.utils.basic_utils import \
|
17 |
pack_imagelist_into_tensor_v2 as pack_imagelist_into_tensor
|
|
|
11 |
import torchvision.transforms.functional as TF
|
12 |
from PIL import Image
|
13 |
import torchvision.transforms as T
|
14 |
+
from scepter.modules.model.registry import DIFFUSIONS,BACKBONES
|
15 |
from scepter.modules.model.utils.basic_utils import check_list_of_list
|
16 |
from scepter.modules.model.utils.basic_utils import \
|
17 |
pack_imagelist_into_tensor_v2 as pack_imagelist_into_tensor
|