tolgacangoz
commited on
Upload matryoshka.py
Browse files- 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]
|
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 |
|