Test TRELLIS models, still waiting for spconv to fix its error for python3.12 cu124

https://github.com/traveller59/spconv/issues/731
This commit is contained in:
Mr. For Example
2024-12-10 19:24:22 -06:00
parent 1def79b136
commit 0641e22a65
19 changed files with 210 additions and 37 deletions
+1 -1
View File
@@ -13,7 +13,7 @@ libs_dir_name: "_Libs"
# Folder used to store built wheels
wheels_dir_name: "_Build_Wheels"
build_base_packages: ["torch", "torchvision"]
build_base_packages: ["torch", "torchvision", "xformers", "spconv-${cuda_version}"]
# All packages will be automatically installed at beginning of install.py follwing orders from top to bottom
remote_packages:
torch:
+14 -9
View File
@@ -140,12 +140,17 @@ def git_folder_parallel(repo_id: str, folder: str, recursive: bool=True, root_ou
def install_remote_packages(package_names):
for package_name in package_names:
package_attr = build_config.remote_packages[package_name]
if hasattr(package_attr, "version"):
package_name += f"=={package_attr.version}"
url_option = package_attr.url_option if hasattr(package_attr, "url_option") else "--index-url"
subprocess.run([
PYTHON_PATH, "-s", "-m", "pip", "install",
package_name, url_option, package_attr.url
])
if package_name in build_config.remote_packages:
package_attr = build_config.remote_packages[package_name]
if hasattr(package_attr, "version"):
package_name += f"=={package_attr.version}"
if hasattr(package_attr, "url"):
url_option = package_attr.url_option if hasattr(package_attr, "url_option") else "--index-url"
subprocess.run([
PYTHON_PATH, "-s", "-m", "pip", "install",
package_name, url_option, package_attr.url
])
continue
subprocess.run([PYTHON_PATH, "-s", "-m", "pip", "install", package_name])
+2 -1
View File
@@ -2,4 +2,5 @@ https://github.com/ashawkey/diff-gaussian-rasterization.git
https://github.com/NVlabs/nvdiffrast.git
https://github.com/ashawkey/kiuikit.git
https://github.com/facebookresearch/pytorch3d.git
https://github.com/rusty1s/pytorch_scatter.git
https://github.com/rusty1s/pytorch_scatter.git
https://github.com/EasternJournalist/utils3d.git