JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-06-25 11:39:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #774 from emmanuel-ferdman/main

Browse Source 
Resolve Python Logger warnings
This commit is contained in:
Yuxuan Zhang 2025-06-17 10:08:33 +08:00 committed by GitHub
commit aaab2877ec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
sat/sgm/models/autoencoder.py
View File

@ -155,7 +155,7 @@ class AutoencodingEngine(AbstractAutoencoder):
) )
assert len(self.ae_optimizer_args) == len(self.trainable_ae_params) assert len(self.ae_optimizer_args) == len(self.trainable_ae_params)
else: else:
self.ae_optimizer_args = [{}] # makes type consitent self.ae_optimizer_args = [{}] # makes type consistent
self.trainable_disc_params = trainable_disc_params self.trainable_disc_params = trainable_disc_params
if self.trainable_disc_params is not None: if self.trainable_disc_params is not None:
@ -165,11 +165,11 @@ class AutoencodingEngine(AbstractAutoencoder):
) )
assert len(self.disc_optimizer_args) == len(self.trainable_disc_params) assert len(self.disc_optimizer_args) == len(self.trainable_disc_params)
else: else:
self.disc_optimizer_args = [{}] # makes type consitent self.disc_optimizer_args = [{}] # makes type consistent
if ckpt_path is not None: if ckpt_path is not None:
assert ckpt_engine is None, "Can't set ckpt_engine and ckpt_path" assert ckpt_engine is None, "Can't set ckpt_engine and ckpt_path"
logpy.warn("Checkpoint path is deprecated, use `checkpoint_egnine` instead") logpy.warning("Checkpoint path is deprecated, use `checkpoint_egnine` instead")
self.apply_ckpt(default(ckpt_path, ckpt_engine)) self.apply_ckpt(default(ckpt_path, ckpt_engine))
self.additional_decode_keys = set(default(additional_decode_keys, [])) self.additional_decode_keys = set(default(additional_decode_keys, []))
@ -358,7 +358,7 @@ class AutoencodingEngine(AbstractAutoencoder):
pattern_params.append(param) pattern_params.append(param)
num_params += param.numel() num_params += param.numel()
if len(pattern_params) == 0: if len(pattern_params) == 0:
logpy.warn(f"Did not find parameters for pattern {pattern_}") logpy.warning(f"Did not find parameters for pattern {pattern_}")
params.extend(pattern_params) params.extend(pattern_params)
groups.append({"params": params, **args}) groups.append({"params": params, **args})
return groups, num_params return groups, num_params

8
sat/vae_modules/autoencoder.py
View File

@ -149,7 +149,7 @@ class AutoencodingEngine(AbstractAutoencoder):
) )
assert len(self.ae_optimizer_args) == len(self.trainable_ae_params) assert len(self.ae_optimizer_args) == len(self.trainable_ae_params)
else: else:
self.ae_optimizer_args = [{}] # makes type consitent self.ae_optimizer_args = [{}] # makes type consistent
self.trainable_disc_params = trainable_disc_params self.trainable_disc_params = trainable_disc_params
if self.trainable_disc_params is not None: if self.trainable_disc_params is not None:
@ -159,11 +159,11 @@ class AutoencodingEngine(AbstractAutoencoder):
) )
assert len(self.disc_optimizer_args) == len(self.trainable_disc_params) assert len(self.disc_optimizer_args) == len(self.trainable_disc_params)
else: else:
self.disc_optimizer_args = [{}] # makes type consitent self.disc_optimizer_args = [{}] # makes type consistent
if ckpt_path is not None: if ckpt_path is not None:
assert ckpt_engine is None, "Can't set ckpt_engine and ckpt_path" assert ckpt_engine is None, "Can't set ckpt_engine and ckpt_path"
logpy.warn("Checkpoint path is deprecated, use `checkpoint_egnine` instead") logpy.warning("Checkpoint path is deprecated, use `checkpoint_egnine` instead")
self.apply_ckpt(default(ckpt_path, ckpt_engine)) self.apply_ckpt(default(ckpt_path, ckpt_engine))
self.additional_decode_keys = set(default(additional_decode_keys, [])) self.additional_decode_keys = set(default(additional_decode_keys, []))
@ -352,7 +352,7 @@ class AutoencodingEngine(AbstractAutoencoder):
pattern_params.append(param) pattern_params.append(param)
num_params += param.numel() num_params += param.numel()
if len(pattern_params) == 0: if len(pattern_params) == 0:
logpy.warn(f"Did not find parameters for pattern {pattern_}") logpy.warning(f"Did not find parameters for pattern {pattern_}")
params.extend(pattern_params) params.extend(pattern_params)
groups.append({"params": params, **args}) groups.append({"params": params, **args})
return groups, num_params return groups, num_params