fix bugs

.gitignore

@@ -0,0 +1,5 @@
+# pycache
+__pycache__/
+
+# ignore sdxl
+pretrained-models/*.safetensors

Dockerfile

@@ -26,7 +26,6 @@ ENV HOME=/home/user \
 	SYSTEM=spaces
 
 # Install miniconda
-RUN mkdir -p /home/user/conda 
 ENV CONDA_DIR /home/user/conda
 RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
     /bin/bash ~/miniconda.sh -b -p /home/user/conda

app.py

@@ -9,6 +9,7 @@ from PIL import Image
 import time
 import tqdm
 import copy
+import sys
 
 # Mesh imports
 from pytorch3d.io import load_objs_as_meshes
@@ -17,6 +18,8 @@ from pytorch3d.transforms import Transform3d, RotateAxisAngle, Translate, Rotate
 
 from sampling_for_demo import load_and_return_model_and_data, sample, load_base_model
 
+# add current directory to path
+# sys.path.append(os.path.dirname(os.path.realpath(__file__)))
 
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 

pretrained-models/car0/camera.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1fae49a540a313275e795d45b994ed3846f7d443e725bb883881c413dfa95fe2
+size 147296

pretrained-models/chair191/camera.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:394dbaf6c87bc98018a2f30c3e901acb3868959065624fc7588986c23abd54b4
+size 147296

pretrained-models/motorcycle12/camera.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f01c0aaf6a65a6bbe4e4a4e5e028e15e7ebe1bda5db65dd887e55c82b4b141b0
+size 147296

pretrained-models/teddybear31/camera.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5e316b57efb602934896dbea3b4ae394320e21c5513fa89beee6cd09adb2447
+size 147296

sampling_for_demo.py

@@ -21,15 +21,6 @@ from sgm.util import instantiate_from_config, load_safetensors
 
 choices = []
 
-def append_dims(x, target_dims):
-    """Appends dimensions to the end of a tensor until it has target_dims dimensions."""
-    dims_to_append = target_dims - x.ndim
-    if dims_to_append < 0:
-        raise ValueError(
-            f"input has {x.ndim} dims but target_dims is {target_dims}, which is less"
-        )
-    return x[(...,) + (None,) * dims_to_append]
-
 
 def load_base_model(config, ckpt=None, verbose=True):
     config = OmegaConf.load(config)