From 82f15b30f905c04fcbf8fe33d859f3c3f72e8fc6 Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Tue, 7 May 2024 18:34:22 +0800 Subject: [PATCH] fixed checkpoint (#92) --- opensora/models/stdit/stdit3.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/opensora/models/stdit/stdit3.py b/opensora/models/stdit/stdit3.py index b7de22f..4bec074 100644 --- a/opensora/models/stdit/stdit3.py +++ b/opensora/models/stdit/stdit3.py @@ -164,7 +164,6 @@ class STDiT3Config(PretrainedConfig): drop_path=0.0, caption_channels=4096, model_max_length=300, - dtype=torch.float32, qk_norm=False, enable_flash_attn=False, enable_layernorm_kernel=False, @@ -185,7 +184,6 @@ class STDiT3Config(PretrainedConfig): self.drop_path = drop_path self.caption_channels = caption_channels self.model_max_length = model_max_length - self.dtype = dtype self.qk_norm = qk_norm self.enable_flash_attn = enable_flash_attn self.enable_layernorm_kernel = enable_layernorm_kernel