Skip to content

Commit 9c3a81d

Browse files
remove None and TODO
1 parent 53ff253 commit 9c3a81d

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

scripts/convert_cogview3_to_diffusers.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,7 @@ def convert_cogview3_vae_checkpoint_to_diffusers(ckpt_path, vae_config):
156156

157157

158158
def main(args):
159-
if args.dtype is None:
160-
dtype = None
159+
161160
if args.dtype == "fp16":
162161
dtype = torch.float16
163162
elif args.dtype == "bf16":
@@ -212,7 +211,6 @@ def main(args):
212211
for param in text_encoder.parameters():
213212
param.data = param.data.contiguous()
214213

215-
# TODO: figure out the correct scheduler if it is same as CogVideoXDDIMScheduler
216214
scheduler = CogVideoXDDIMScheduler.from_config(
217215
{
218216
"snr_shift_scale": 4.0,

0 commit comments

Comments
 (0)