Spaces:
Running
Running
Antoni Bigata
commited on
Commit
·
34f5ce5
1
Parent(s):
afe1949
requirements
Browse files
sgm/modules/encoders/modules.py
CHANGED
|
@@ -6,7 +6,11 @@ import os
|
|
| 6 |
|
| 7 |
import kornia
|
| 8 |
import numpy as np
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
-
|
| 40 |
-
from
|
|
|
|
|
|
|
|
|
|
| 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):
|