Update briacustom.py
Browse files- briacustom.py +1 -1
briacustom.py
CHANGED
|
@@ -3,7 +3,7 @@ import torch.nn as nn
|
|
| 3 |
import torch.nn.functional as F
|
| 4 |
from typing import Dict, List, Tuple, Optional
|
| 5 |
from enum import Enum, auto
|
| 6 |
-
from
|
| 7 |
class ClothingType(Enum):
|
| 8 |
SHIRT = auto()
|
| 9 |
PANTS = auto()
|
|
|
|
| 3 |
import torch.nn.functional as F
|
| 4 |
from typing import Dict, List, Tuple, Optional
|
| 5 |
from enum import Enum, auto
|
| 6 |
+
from briarmbg import advanced_briarmb
|
| 7 |
class ClothingType(Enum):
|
| 8 |
SHIRT = auto()
|
| 9 |
PANTS = auto()
|