From 8d1d6aa9679ac191b7fe58b9e124579c19ae612e Mon Sep 17 00:00:00 2001 From: Alexandros Koumparoulis Date: Tue, 22 Oct 2024 14:05:08 -0700 Subject: [PATCH] copyright Signed-off-by: Alexandros Koumparoulis --- examples/llm/sft/hf.py | 14 ++++++++++++++ nemo/collections/llm/gpt/data/hf_dataset.py | 14 ++++++++++++++ .../llm/gpt/model/hf_auto_model_for_causal_lm.py | 14 ++++++++++++++ 3 files changed, 42 insertions(+) diff --git a/examples/llm/sft/hf.py b/examples/llm/sft/hf.py index a909527625f4..fbbe0d5eba51 100644 --- a/examples/llm/sft/hf.py +++ b/examples/llm/sft/hf.py @@ -1,3 +1,17 @@ +# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved. +# +# 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. + import fiddle as fdl import pytorch_lightning as pl from pytorch_lightning.loggers import WandbLogger diff --git a/nemo/collections/llm/gpt/data/hf_dataset.py b/nemo/collections/llm/gpt/data/hf_dataset.py index aee0edd8c016..083f5fbc5235 100644 --- a/nemo/collections/llm/gpt/data/hf_dataset.py +++ b/nemo/collections/llm/gpt/data/hf_dataset.py @@ -1,3 +1,17 @@ +# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved. +# +# 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. + import pytorch_lightning as pl import torch from torch.utils.data import DataLoader diff --git a/nemo/collections/llm/gpt/model/hf_auto_model_for_causal_lm.py b/nemo/collections/llm/gpt/model/hf_auto_model_for_causal_lm.py index c5ead0100f23..5833db26856e 100644 --- a/nemo/collections/llm/gpt/model/hf_auto_model_for_causal_lm.py +++ b/nemo/collections/llm/gpt/model/hf_auto_model_for_causal_lm.py @@ -1,3 +1,17 @@ +# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved. +# +# 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. + import pytorch_lightning as pl import torch import torch.nn.functional as F