jadechoghari commited on
Commit
f691fcb
·
verified ·
1 Parent(s): 4aa2be3

Update modeling.py

Browse files
Files changed (1) hide show
  1. modeling.py +6 -0
modeling.py CHANGED
@@ -8,6 +8,12 @@ from .lrm.models.encoders.dino_wrapper2 import DinoWrapper
8
  from .transformer import TriplaneTransformer
9
  from .lrm.models.rendering.synthesizer_part import TriplaneSynthesizer
10
 
 
 
 
 
 
 
11
  class CameraEmbedder(nn.Module):
12
  def __init__(self, raw_dim: int, embed_dim: int):
13
  super().__init__()
 
8
  from .transformer import TriplaneTransformer
9
  from .lrm.models.rendering.synthesizer_part import TriplaneSynthesizer
10
 
11
+ # debug
12
+ package = __import__('lrm.models.rendering', fromlist=[''])
13
+ print("available modules in lrm.models.rendering:")
14
+ for importer, modname, ispkg in pkgutil.iter_modules(package.__path__):
15
+ print(modname)
16
+
17
  class CameraEmbedder(nn.Module):
18
  def __init__(self, raw_dim: int, embed_dim: int):
19
  super().__init__()