Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Compatible with Jetson AGX Orin #48

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Open

Compatible with Jetson AGX Orin #48

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
f855ef8
Merge pull request #46 from Lyken17/main
Lyken17 Nov 26, 2024
9110d56
update README.md.
xieenze Nov 26, 2024
1808a02
Update pyproject.toml
johnnynunez Nov 26, 2024
b1be116
Update pyproject.toml
johnnynunez Nov 26, 2024
d3f1c7e
1. update app
lawrence-cj Nov 26, 2024
a1d1e14
pre-commit;
lawrence-cj Nov 26, 2024
94e7733
add a AdamW optimizer type as config file for reference.
lawrence-cj Nov 26, 2024
7d0d659
fix all the model input z.to(weight_dtype) bugs;
lawrence-cj Nov 26, 2024
34bef75
Merge branch 'main' into patch-1
lawrence-cj Nov 26, 2024
386dcf6
Merge branch 'main' into patch-1
johnnynunez Nov 28, 2024
45b3538
Merge branch 'NVlabs:main' into patch-1
johnnynunez Dec 20, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -350,4 +350,4 @@ We will try our best to release
primaryClass={cs.CV},
url={https://arxiv.org/abs/2410.10629},
}
```
```
10 changes: 5 additions & 5 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ dependencies = [
"gradio",
"image-reward",
"ipdb",
"mmcv==1.7.2",
"mmcv==2.2.0",
"omegaconf",
"opencv-python",
"optimum",
Expand All @@ -38,13 +38,13 @@ dependencies = [
"tensorboard",
"tensorboardX",
"timm",
"torchaudio==2.4.0",
"torchvision==0.19",
"torchaudio==2.5.0",
"torchvision==0.20.0",
"transformers",
"triton==3.0.0",
"triton==3.1.0",
"wandb",
"webdataset",
"xformers==0.0.27.post2",
"xformers",
"yapf",
"spaces",
"matplotlib",
Expand Down