fix

Author: Eigensystem

fastvideo/distill/solver.py

@@ -248,11 +248,6 @@ def __init__(self, sigmas, timesteps=1000, euler_timesteps=50):
         self.sigmas = sigmas[self.euler_timesteps]
         self.sigmas_prev = np.asarray([sigmas[0]] +
                                       sigmas[self.euler_timesteps[:-1]].tolist())  # either use sigma0 or 0
-        print(f"sigmas: {sigmas}")
-        print(f"euler_timesteps: {self.euler_timesteps}")
-        print(f"sigmas: {self.sigmas}")
-        print(f"sigmas_prev: {self.sigmas_prev}")
-
         self.euler_timesteps = torch.from_numpy(self.euler_timesteps).long()
         self.euler_timesteps_prev = torch.from_numpy(self.euler_timesteps_prev).long()
         self.sigmas = torch.from_numpy(self.sigmas)

fastvideo/v1/models/loader/component_loader.py

@@ -367,7 +367,7 @@ class TransformerLoader(ComponentLoader):
     def load(self, model_path: str, architecture: str,
              fastvideo_args: FastVideoArgs):
         """Load the transformer based on the model path, architecture, and inference args."""
-        print(f"Loading transformer from {model_path}")
+        logger.info("Loading transformer from %s", model_path)
         config = get_diffusers_config(model=model_path)
         hf_config = deepcopy(config)
         cls_name = config.pop("_class_name")