From 34711b7bd286e3c5fb0fd0d6aa4c70fefeabb54d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=91=A8=E9=92=B0=E5=9D=A4?= <90625606+1zeryu@users.noreply.github.com> Date: Tue, 30 Apr 2024 13:37:25 +0800 Subject: [PATCH] update --- opensora/models/pixart/pixart.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/opensora/models/pixart/pixart.py b/opensora/models/pixart/pixart.py index 87cf496..b5bd9ca 100644 --- a/opensora/models/pixart/pixart.py +++ b/opensora/models/pixart/pixart.py @@ -402,14 +402,3 @@ def PixArtMS_XL_2(from_pretrained=None, **kwargs): if from_pretrained is not None: load_checkpoint(model, from_pretrained) return model -<<<<<<< Updated upstream -======= - - -@MODELS.register_module("PixArt-1B/2") -def PixArt_1B_2(from_pretrained=None, **kwargs): - model = PixArt(depth=28, hidden_size=1872, patch_size=(1, 2, 2), num_heads=26, **kwargs) - if from_pretrained is not None: - load_checkpoint(model, from_pretrained) - return model ->>>>>>> Stashed changes