From 57e7b21e59f53d99a422c22694c3557073e42ebd Mon Sep 17 00:00:00 2001 From: GitLab Mirror Bot Date: Wed, 12 Feb 2025 01:32:55 +0000 Subject: [PATCH] Stability improvements. --- cosmos1/__init__.py | 14 ++++++++++++++ cosmos1/models/__init__.py | 14 ++++++++++++++ cosmos1/models/tokenizer/nemo/README.md | 3 +++ cosmos1/scripts/__init__.py | 14 ++++++++++++++ cosmos1/utils/__init__.py | 14 ++++++++++++++ requirements.txt | 2 +- 6 files changed, 60 insertions(+), 1 deletion(-) create mode 100644 cosmos1/__init__.py create mode 100644 cosmos1/models/__init__.py create mode 100644 cosmos1/scripts/__init__.py create mode 100644 cosmos1/utils/__init__.py diff --git a/cosmos1/__init__.py b/cosmos1/__init__.py new file mode 100644 index 0000000..dac9a4d --- /dev/null +++ b/cosmos1/__init__.py @@ -0,0 +1,14 @@ +# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/cosmos1/models/__init__.py b/cosmos1/models/__init__.py new file mode 100644 index 0000000..dac9a4d --- /dev/null +++ b/cosmos1/models/__init__.py @@ -0,0 +1,14 @@ +# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/cosmos1/models/tokenizer/nemo/README.md b/cosmos1/models/tokenizer/nemo/README.md index cf05e5a..7c8b9fc 100644 --- a/cosmos1/models/tokenizer/nemo/README.md +++ b/cosmos1/models/tokenizer/nemo/README.md @@ -86,6 +86,9 @@ Complete the following steps to post-train the Cosmos tokenizer Cosmos-1.0-Token pip install megatron-energon==4.0.0 pyav pip install git+https://github.com/NVIDIA/NeMo-Run.git pip install moviepy==1.0.3 imageio + + # switch to NeMo branch supporting tokenizer post-training + cd /opt/NeMo && git fetch origin cosmos_tokenizer && git checkout cosmos_tokenizer ``` 2. Run the following command to post-train Cosmos-1.0-Tokenizer-CV8x8x8: diff --git a/cosmos1/scripts/__init__.py b/cosmos1/scripts/__init__.py new file mode 100644 index 0000000..dac9a4d --- /dev/null +++ b/cosmos1/scripts/__init__.py @@ -0,0 +1,14 @@ +# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/cosmos1/utils/__init__.py b/cosmos1/utils/__init__.py new file mode 100644 index 0000000..dac9a4d --- /dev/null +++ b/cosmos1/utils/__init__.py @@ -0,0 +1,14 @@ +# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/requirements.txt b/requirements.txt index 81394b8..c4cdcf1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -18,7 +18,6 @@ av better-profanity einops==0.7.0 einx==0.1.3 -git+https://github.com/NVlabs/Pytorch_Retinaface.git@b843f45 huggingface-hub>=0.26.2 hydra-core imageio[ffmpeg] @@ -28,6 +27,7 @@ mediapy nltk peft pillow +pytorch_retinaface @ git+https://github.com/NVlabs/Pytorch_Retinaface.git@b843f45 sentencepiece termcolor transformers==4.45.0