chaojiemao commited on
Commit
08e3efa
·
verified ·
1 Parent(s): 532ddca

Update ace_inference.py

Browse files
Files changed (1) hide show
  1. 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