Hugging Face's logo

jadechoghari
/
vfusion3d

Image-to-3D
Transformers
Safetensors
lrm_generator
feature-extraction
custom_code
Model card Files Community
7
jadechoghari commited on
Commit
c577b6d
·
verified ·
1 Parent(s): c4d78ab

update lrm wrong import

Browse files
Files changed (1) hide show
  1. modeling.py +3 -3
modeling.py CHANGED
@@ -4,9 +4,9 @@ import torch.nn as nn
4
  from transformers import PreTrainedModel, PretrainedConfig
5
  import torch
6
  # import dinowrapper
7
- from lrm.models.encoders.dino_wrapper2 import DinoWrapper
8
- from lrm.models.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):
 
4
  from transformers import PreTrainedModel, PretrainedConfig
5
  import torch
6
  # import dinowrapper
7
+ from .lrm.models.encoders.dino_wrapper2 import DinoWrapper
8
+ from .lrm.models.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):