tolgacangoz commited on
Commit
e30e6fa
·
verified ·
1 Parent(s): f697d2b

Upload matryoshka.py

Browse files
Files changed (1) hide show
  1. matryoshka.py +2 -1
matryoshka.py CHANGED
@@ -3750,6 +3750,7 @@ class MatryoshkaPipeline(
3750
  scheduler: MatryoshkaDDIMScheduler,
3751
  feature_extractor: CLIPImageProcessor = None,
3752
  image_encoder: CLIPVisionModelWithProjection = None,
 
3753
  ):
3754
  super().__init__()
3755
 
@@ -4510,7 +4511,7 @@ class MatryoshkaPipeline(
4510
  timesteps, num_inference_steps = retrieve_timesteps(
4511
  self.scheduler, num_inference_steps, device, timesteps, sigmas
4512
  )
4513
- timesteps = timesteps[:-1] # is this correct???
4514
  else:
4515
  timesteps = self.scheduler.timesteps
4516
 
 
3750
  scheduler: MatryoshkaDDIMScheduler,
3751
  feature_extractor: CLIPImageProcessor = None,
3752
  image_encoder: CLIPVisionModelWithProjection = None,
3753
+ trust_remote_code: bool = False,
3754
  ):
3755
  super().__init__()
3756
 
 
4511
  timesteps, num_inference_steps = retrieve_timesteps(
4512
  self.scheduler, num_inference_steps, device, timesteps, sigmas
4513
  )
4514
+ timesteps = timesteps[:-1]
4515
  else:
4516
  timesteps = self.scheduler.timesteps
4517