Skip to content

Commit

Permalink
Update test_consistency_models.py
Browse files Browse the repository at this point in the history
rerun ci check
  • Loading branch information
xiaomile committed Dec 14, 2023
1 parent f68dc99 commit 015e5d4
Showing 1 changed file with 0 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,6 @@ def test_init(self):
self.assertIsInstance(model.data_preprocessor, DataPreprocessor)
self.assertIsInstance(model.model, ConsistencyUNetModel)
self.assertIsInstance(model.diffusion, KarrasDenoiser)

unet_cfg = deepcopy(unet_config)
diffuse_cfg = deepcopy(denoiser_config)
unet = MODELS.build(unet_cfg)
Expand Down

0 comments on commit 015e5d4

Please sign in to comment.