xrg
commited on
Commit
·
b7b3119
1
Parent(s):
037d383
fix bug
Browse files- app.py +1 -1
- core/embedder.py +0 -1
- 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 |
|