Antoni Bigata commited on
Commit
34f5ce5
·
1 Parent(s): afe1949

requirements

Browse files
Files changed (1) hide show
  1. sgm/modules/encoders/modules.py +10 -3
sgm/modules/encoders/modules.py CHANGED
@@ -6,7 +6,11 @@ import os
6
 
7
  import kornia
8
  import numpy as np
9
- import open_clip
 
 
 
 
10
  import torch
11
  import torch.nn as nn
12
  from einops import rearrange, repeat
@@ -36,8 +40,11 @@ from ...util import (
36
  instantiate_from_config,
37
  )
38
 
39
- from facenet_pytorch import MTCNN, InceptionResnetV1
40
- from insightface.app import FaceAnalysis
 
 
 
41
 
42
 
43
  class AbstractEmbModel(nn.Module):
 
6
 
7
  import kornia
8
  import numpy as np
9
+
10
+ try:
11
+ import open_clip
12
+ except ImportError:
13
+ pass
14
  import torch
15
  import torch.nn as nn
16
  from einops import rearrange, repeat
 
40
  instantiate_from_config,
41
  )
42
 
43
+ try:
44
+ from facenet_pytorch import MTCNN, InceptionResnetV1
45
+ from insightface.app import FaceAnalysis
46
+ except ImportError:
47
+ pass
48
 
49
 
50
  class AbstractEmbModel(nn.Module):