xrg commited on
Commit
b7b3119
·
1 Parent(s): 037d383
Files changed (3) hide show
  1. app.py +1 -1
  2. core/embedder.py +0 -1
  3. core/encoders/dino_wrapper.py +0 -1
app.py CHANGED
@@ -33,7 +33,7 @@ GRADIO_OBJ_ALBEDO_PATH = 'gradio_output_albedo.obj'
33
  GRADIO_OBJ_SHADING_PATH = 'gradio_output_shading.obj'
34
 
35
  #opt = tyro.cli(AllConfigs)
36
- torch._dynamo.config.verbose=True
37
 
38
  ckpt_path = hf_hub_download(repo_id="rgxie/LDM", filename="LDM6v01.ckpt")
39
 
 
33
  GRADIO_OBJ_SHADING_PATH = 'gradio_output_shading.obj'
34
 
35
  #opt = tyro.cli(AllConfigs)
36
+ #torch._dynamo.config.verbose=True
37
 
38
  ckpt_path = hf_hub_download(repo_id="rgxie/LDM", filename="LDM6v01.ckpt")
39
 
core/embedder.py CHANGED
@@ -32,6 +32,5 @@ class CameraEmbedder(nn.Module):
32
  nn.Linear(embed_dim, embed_dim),
33
  )
34
 
35
- @torch.compile
36
  def forward(self, x):
37
  return self.mlp(x)
 
32
  nn.Linear(embed_dim, embed_dim),
33
  )
34
 
 
35
  def forward(self, x):
36
  return self.mlp(x)
core/encoders/dino_wrapper.py CHANGED
@@ -32,7 +32,6 @@ class DinoWrapper(nn.Module):
32
  if freeze:
33
  self._freeze()
34
 
35
- @torch.compile
36
  def forward_model(self, inputs):
37
  return self.model(**inputs, interpolate_pos_encoding=True)
38
 
 
32
  if freeze:
33
  self._freeze()
34
 
 
35
  def forward_model(self, inputs):
36
  return self.model(**inputs, interpolate_pos_encoding=True)
37