nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 13:14:44 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

update

Browse source
This commit is contained in:
周钰坤 2024-04-30 13:37:25 +08:00
parent 59591d8afc
commit 34711b7bd2
1 changed files with 0 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
opensora/models/pixart/pixart.py
View file

@ -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