Update modeling.py
Browse files- modeling.py +3 -3
modeling.py
CHANGED
|
@@ -6,9 +6,9 @@ import torch
|
|
| 6 |
import numpy as np
|
| 7 |
import math
|
| 8 |
|
| 9 |
-
from dino_wrapper2 import DinoWrapper
|
| 10 |
-
from transformer import TriplaneTransformer
|
| 11 |
-
from synthesizer_part import TriplaneSynthesizer
|
| 12 |
|
| 13 |
class CameraEmbedder(nn.Module):
|
| 14 |
def __init__(self, raw_dim: int, embed_dim: int):
|
|
|
|
| 6 |
import numpy as np
|
| 7 |
import math
|
| 8 |
|
| 9 |
+
from .dino_wrapper2 import DinoWrapper
|
| 10 |
+
from .transformer import TriplaneTransformer
|
| 11 |
+
from .synthesizer_part import TriplaneSynthesizer
|
| 12 |
|
| 13 |
class CameraEmbedder(nn.Module):
|
| 14 |
def __init__(self, raw_dim: int, embed_dim: int):
|