update app.py

app.py

@@ -58,10 +58,8 @@ if os.path.exists(local_ckpt):
     ckpt = local_ckpt
 else:
     ckpt = hf_hub_download(repo_id="hongfz16/3DTopia", filename="model.safetensors")
-print("download finish")
 configs = OmegaConf.load(config)
 os.makedirs("tmp", exist_ok=True)
-print("download finish")
 
 import sys
 import traceback
@@ -70,7 +68,6 @@ try:
     if ckpt.endswith(".ckpt"):
         model = get_obj_from_str(configs.model["target"]).load_from_checkpoint(ckpt, map_location='cpu', strict=False, **configs.model.params)
     elif ckpt.endswith(".safetensors"):
-        print("download finish")
         model = get_obj_from_str(configs.model["target"])(**configs.model.params)
         print("download finish")
         model_ckpt = load_file(ckpt)

requirements.txt

@@ -1,58 +1,57 @@
---extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.12.0+cu113
-torchvision==0.13.0+cu113
-torchaudio==0.12.0
+torch
+torchvision
+torchaudio
 pytorch-lightning
 numpy
 tqdm
 PyYAML
 git+https://github.com/openai/CLIP.git
-einops==0.6.1
-huggingface-hub==0.16.4
-imageio==2.31.0
-imageio-ffmpeg==0.4.8
-importlib-resources==6.1.0
-ipdb==0.13.13
-ipython==8.12.2
-jedi==0.19.0
-kiwisolver==1.4.5
-kornia==0.6.0
-lpips==0.1.4
-matplotlib==3.7.3
-matplotlib-inline==0.1.6
-omegaconf==2.3.0
-open-clip-torch==2.20.0
-opencv-python==4.7.0.72
-parso==0.8.3
-pathtools==0.1.2
-pexpect==4.8.0
-pickleshare==0.7.5
-pillow==9.5.0
-prompt-toolkit==3.0.39
-protobuf==3.20.3
-psutil==5.9.5
-ptyprocess==0.7.0
-pure-eval==0.2.2
-pygments==2.16.1
-pymcubes==0.1.4
-pyparsing==3.1.1
-pytorch-fid==0.3.0
-pytorch-msssim==1.0.0
-regex==2023.6.3
-safetensors==0.3.3
-scipy==1.10.1
-sentencepiece==0.1.99
-sentry-sdk==1.25.0
-setproctitle==1.3.2
-smmap==5.0.0
-stack-data==0.6.2
-timm==0.9.7
-tokenizers==0.12.1
-tomli==2.0.1
-traitlets==5.9.0
+einops
+huggingface-hub
+imageio
+imageio-ffmpeg
+importlib-resources
+ipdb
+ipython
+jedi
+kiwisolver
+kornia
+lpips
+matplotlib
+matplotlib-inline
+omegaconf
+open-clip-torch
+opencv-python
+parso
+pathtools
+pexpect
+pickleshare
+pillow
+prompt-toolkit
+protobuf
+psutil
+ptyprocess
+pure-eval
+pygments
+pymcubes
+pyparsing
+pytorch-fid
+pytorch-msssim
+regex
+safetensors
+scipy
+sentencepiece
+sentry-sdk
+setproctitle
+smmap
+stack-data
+timm
+tokenizers
+tomli
+traitlets
 transformers
-trimesh==4.0.2
-vit-pytorch==1.2.2
-wandb==0.15.3
-wcwidth==0.2.6
-zipp==3.17.0
+trimesh
+vit-pytorch
+wandb
+wcwidth
+zipp