From feab005a0f69f4339fc5e0ac3e9ca459b1c76ba7 Mon Sep 17 00:00:00 2001 From: Jackson Searle Date: Wed, 20 Mar 2024 18:06:55 -0600 Subject: [PATCH] renames longnet file; longnet example in readme works now --- examples/fairseq/models/language_modeling.py | 2 +- examples/longvit/longvit.py | 2 +- examples/longvit/pretraining/vision_transformer.py | 2 +- torchscale/model/{LongNet.py => longnet.py} | 0 4 files changed, 3 insertions(+), 3 deletions(-) rename torchscale/model/{LongNet.py => longnet.py} (100%) diff --git a/examples/fairseq/models/language_modeling.py b/examples/fairseq/models/language_modeling.py index 38cf9e2b..206de181 100644 --- a/examples/fairseq/models/language_modeling.py +++ b/examples/fairseq/models/language_modeling.py @@ -25,7 +25,7 @@ from torchscale.architecture.config import DecoderConfig from torchscale.architecture.decoder import Decoder -from torchscale.model.LongNet import LongNetDecoder +from torchscale.model.longnet import LongNetDecoder DEFAULT_MAX_TARGET_POSITIONS = 1024 logger = logging.getLogger(__name__) diff --git a/examples/longvit/longvit.py b/examples/longvit/longvit.py index 58db485b..322e250e 100644 --- a/examples/longvit/longvit.py +++ b/examples/longvit/longvit.py @@ -23,7 +23,7 @@ import torch.nn as nn from torchscale.architecture.encoder import Encoder -from torchscale.model.LongNet import LongNetEncoder +from torchscale.model.longnet import LongNetEncoder from torchscale.architecture.config import EncoderConfig from timm.models.layers import trunc_normal_ as __call_trunc_normal_ diff --git a/examples/longvit/pretraining/vision_transformer.py b/examples/longvit/pretraining/vision_transformer.py index 675ee8a9..0eb0cf0d 100644 --- a/examples/longvit/pretraining/vision_transformer.py +++ b/examples/longvit/pretraining/vision_transformer.py @@ -23,7 +23,7 @@ from utils import trunc_normal_ from torchscale.architecture.encoder import Encoder -from torchscale.model.LongNet import LongNetEncoder +from torchscale.model.longnet import LongNetEncoder from torchscale.architecture.config import EncoderConfig diff --git a/torchscale/model/LongNet.py b/torchscale/model/longnet.py similarity index 100% rename from torchscale/model/LongNet.py rename to torchscale/model/longnet.py