Huiwenshi commited on
Commit
a13abe3
·
verified ·
1 Parent(s): 53e746c

Upload app_hg.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app_hg.py +4 -5
app_hg.py CHANGED
@@ -60,9 +60,7 @@ def install_requirements():
60
  subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/NVlabs/nvdiffrast"])
61
  subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"])
62
 
63
- # install_cuda_toolkit()
64
- install_requirements()
65
-
66
  def download_models():
67
  os.makedirs("weights", exist_ok=True)
68
  os.makedirs("weights/hunyuanDiT", exist_ok=True)
@@ -95,6 +93,9 @@ def download_models():
95
  except Exception as e:
96
  print(f"Error downloading DUSt3R: {e}")
97
 
 
 
 
98
 
99
  from infer import seed_everything, save_gif
100
  from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer
@@ -122,8 +123,6 @@ parser.add_argument("--save_memory", default=False)
122
  parser.add_argument("--device", default="cuda:0", type=str)
123
  args = parser.parse_args()
124
 
125
- download_models() ### download weights !!!!
126
-
127
  ################################################################
128
  # initial setting
129
  ################################################################
 
60
  subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/NVlabs/nvdiffrast"])
61
  subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"])
62
 
63
+
 
 
64
  def download_models():
65
  os.makedirs("weights", exist_ok=True)
66
  os.makedirs("weights/hunyuanDiT", exist_ok=True)
 
93
  except Exception as e:
94
  print(f"Error downloading DUSt3R: {e}")
95
 
96
+ # install_cuda_toolkit()
97
+ install_requirements()
98
+ download_models() ### download weights !!!!
99
 
100
  from infer import seed_everything, save_gif
101
  from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer
 
123
  parser.add_argument("--device", default="cuda:0", type=str)
124
  args = parser.parse_args()
125
 
 
 
126
  ################################################################
127
  # initial setting
128
  ################################################################