Compare commits

...

1 Commits

Author SHA1 Message Date
Dhruv Nair
b6e47d82dd update 2024-07-24 08:02:47 +00:00
80 changed files with 94 additions and 0 deletions

View File

@@ -58,6 +58,7 @@ class SD3LoRATests(unittest.TestCase):
out_channels=4,
)
clip_text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -83,6 +83,7 @@ class AmusedPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=8,

View File

@@ -87,6 +87,7 @@ class AmusedImg2ImgPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=8,

View File

@@ -87,6 +87,7 @@ class AmusedInpaintPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=8,

View File

@@ -80,6 +80,7 @@ class AnimateDiffPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -93,6 +93,7 @@ class AnimateDiffPipelineSDXLFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -75,6 +75,7 @@ class AnimateDiffVideoToVideoPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -63,6 +63,7 @@ class BlipDiffusionPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
def get_dummy_components(self):
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
vocab_size=1000,
hidden_size=8,
intermediate_size=8,

View File

@@ -167,6 +167,7 @@ class ControlNetPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -354,6 +355,7 @@ class StableDiffusionMultiControlNetPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -574,6 +576,7 @@ class StableDiffusionMultiControlNetOneModelPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -71,6 +71,7 @@ class BlipDiffusionControlNetPipelineFastTests(PipelineTesterMixin, unittest.Tes
def get_dummy_components(self):
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
vocab_size=1000,
hidden_size=16,
intermediate_size=16,

View File

@@ -117,6 +117,7 @@ class ControlNetImg2ImgPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -261,6 +262,7 @@ class StableDiffusionMultiControlNetPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -106,6 +106,7 @@ class ControlNetInpaintPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -223,6 +224,7 @@ class ControlNetSimpleInpaintPipelineFastTests(ControlNetInpaintPipelineFastTest
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -316,6 +318,7 @@ class MultiControlNetInpaintPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -126,6 +126,7 @@ class ControlNetPipelineSDXLFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -130,6 +130,7 @@ class StableDiffusionXLControlNetPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -568,6 +569,7 @@ class StableDiffusionXLMultiControlNetPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -755,6 +757,7 @@ class StableDiffusionXLMultiControlNetOneModelPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -1080,6 +1083,7 @@ class StableDiffusionSSD1BControlNetPipelineFastTests(StableDiffusionXLControlNe
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -118,6 +118,7 @@ class ControlNetPipelineSDXLImg2ImgFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -86,6 +86,7 @@ class StableDiffusion3ControlNetPipelineFastTests(unittest.TestCase, PipelineTes
out_channels=8,
)
clip_text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -180,6 +180,7 @@ class ControlNetXSPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=8,

View File

@@ -119,6 +119,7 @@ class StableDiffusionXLControlNetXSPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=4,

View File

@@ -98,6 +98,7 @@ class I2VGenXLPipelineFastTests(SDFunctionTesterMixin, PipelineTesterMixin, unit
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=4,

View File

@@ -66,6 +66,7 @@ class Dummies:
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,

View File

@@ -67,6 +67,7 @@ class Dummies:
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,

View File

@@ -83,6 +83,7 @@ class KandinskyV22PriorEmb2EmbPipelineFastTests(PipelineTesterMixin, unittest.Te
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,

View File

@@ -68,6 +68,7 @@ class LatentConsistencyModelPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -76,6 +76,7 @@ class LatentConsistencyModelImg2ImgPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -81,6 +81,7 @@ class LDMTextToImagePipelineFastTests(PipelineTesterMixin, unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -70,6 +70,7 @@ class LEditsPPPipelineStableDiffusionFastTests(unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -112,6 +112,7 @@ class LEditsPPPipelineStableDiffusionXLFastTests(unittest.TestCase):
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -96,6 +96,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -96,6 +96,7 @@ class MarigoldNormalsPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -109,6 +109,7 @@ class StableDiffusionControlNetPAGPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=8,

View File

@@ -122,6 +122,7 @@ class StableDiffusionXLControlNetPAGPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -104,6 +104,7 @@ class StableDiffusionPAGPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=cross_attention_dim,

View File

@@ -110,6 +110,7 @@ class StableDiffusionXLPAGPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -149,6 +149,7 @@ class StableDiffusionXLPAGImg2ImgPipelineFastTests(
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -124,6 +124,7 @@ class StableDiffusionXLPAGInpaintPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -83,6 +83,7 @@ class PIAPipelineFastTests(IPAdapterTesterMixin, PipelineTesterMixin, PipelineFr
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -90,6 +90,7 @@ class SafeDiffusionPipelineFastTests(unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -67,6 +67,7 @@ class ShapEPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,

View File

@@ -81,6 +81,7 @@ class StableCascadeCombinedPipelineFastTests(PipelineTesterMixin, unittest.TestC
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
projection_dim=self.text_embedder_hidden_size,

View File

@@ -82,6 +82,7 @@ class StableCascadeDecoderPipelineFastTests(PipelineTesterMixin, unittest.TestCa
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
projection_dim=self.text_embedder_hidden_size,

View File

@@ -87,6 +87,7 @@ class StableCascadePriorPipelineFastTests(PipelineTesterMixin, unittest.TestCase
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,

View File

@@ -159,6 +159,7 @@ class StableDiffusionPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=cross_attention_dim,

View File

@@ -138,6 +138,7 @@ class StableDiffusionImg2ImgPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -140,6 +140,7 @@ class StableDiffusionInpaintPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -427,6 +428,7 @@ class StableDiffusionSimpleInpaintPipelineFastTests(StableDiffusionInpaintPipeli
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -91,6 +91,7 @@ class StableDiffusionInstructPix2PixPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -111,6 +111,7 @@ class StableDiffusion2PipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -109,6 +109,7 @@ class StableDiffusionAttendAndExcitePipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -104,6 +104,7 @@ class StableDiffusionDepth2ImgPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -101,6 +101,7 @@ class StableDiffusionDiffEditPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -84,6 +84,7 @@ class StableDiffusion2InpaintPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -120,6 +120,7 @@ class StableDiffusionLatentUpscalePipelineFastTests(
)
scheduler = EulerDiscreteScheduler(prediction_type="sample")
text_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -97,6 +97,7 @@ class StableDiffusionUpscalePipelineFastTests(unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -92,6 +92,7 @@ class StableDiffusion2VPredictionPipelineFastTests(unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -50,6 +50,7 @@ class StableDiffusion3PipelineFastTests(unittest.TestCase, PipelineTesterMixin):
out_channels=4,
)
clip_text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -52,6 +52,7 @@ class StableDiffusion3Img2ImgPipelineFastTests(PipelineLatentTesterMixin, unitte
out_channels=4,
)
clip_text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -54,6 +54,7 @@ class StableDiffusion3InpaintPipelineFastTests(PipelineLatentTesterMixin, unitte
out_channels=16,
)
clip_text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -82,6 +82,7 @@ class AdapterTests:
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -169,6 +170,7 @@ class AdapterTests:
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -90,6 +90,7 @@ class GligenPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -99,6 +99,7 @@ class GligenTextImagePipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -72,6 +72,7 @@ class StableDiffusionLDM3DPipelineFastTests(unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -81,6 +81,7 @@ class StableDiffusionPanoramaPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -81,6 +81,7 @@ class SafeDiffusionPipelineFastTests(unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -88,6 +88,7 @@ class StableDiffusionSAGPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=8,

View File

@@ -116,6 +116,7 @@ class StableDiffusionXLPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -96,6 +96,7 @@ class StableDiffusionXLAdapterPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -203,6 +204,7 @@ class StableDiffusionXLAdapterPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -141,6 +141,7 @@ class StableDiffusionXLImg2ImgPipelineFastTests(
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,
@@ -563,6 +564,7 @@ class StableDiffusionXLImg2ImgRefinerOnlyPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -111,6 +111,7 @@ class StableDiffusionXLInpaintPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -99,6 +99,7 @@ class StableDiffusionXLInstructPix2PixPipelineFastTests(
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -51,6 +51,7 @@ class StableUnCLIPPipelineFastTests(
torch.manual_seed(0)
prior_text_encoder = CLIPTextModelWithProjection(
CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=embedder_hidden_size,
@@ -94,6 +95,7 @@ class StableUnCLIPPipelineFastTests(
torch.manual_seed(0)
text_encoder = CLIPTextModel(
CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=embedder_hidden_size,

View File

@@ -84,6 +84,7 @@ class StableUnCLIPImg2ImgPipelineFastTests(
torch.manual_seed(0)
text_encoder = CLIPTextModel(
CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=embedder_hidden_size,

View File

@@ -1124,6 +1124,7 @@ class PipelineFastTests(unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -1905,6 +1905,7 @@ class PipelinePushToHubTester(unittest.TestCase):
)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -95,6 +95,7 @@ class TextToVideoSDPipelineFastTests(PipelineTesterMixin, SDFunctionTesterMixin,
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=4,

View File

@@ -100,6 +100,7 @@ class TextToVideoZeroSDXLPipelineFastTests(PipelineTesterMixin, PipelineFromPipe
)
torch.manual_seed(seed)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -105,6 +105,7 @@ class VideoToVideoSDPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
)
torch.manual_seed(0)
text_encoder_config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=32,

View File

@@ -88,6 +88,7 @@ class UnCLIPPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,

View File

@@ -96,6 +96,7 @@ class UnCLIPImageVariationPipelineFastTests(PipelineTesterMixin, unittest.TestCa
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,

View File

@@ -69,6 +69,7 @@ class WuerstchenCombinedPipelineFastTests(PipelineTesterMixin, unittest.TestCase
def dummy_prior_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,
@@ -85,6 +86,7 @@ class WuerstchenCombinedPipelineFastTests(PipelineTesterMixin, unittest.TestCase
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
projection_dim=self.text_embedder_hidden_size,

View File

@@ -70,6 +70,7 @@ class WuerstchenDecoderPipelineFastTests(PipelineTesterMixin, unittest.TestCase)
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
projection_dim=self.text_embedder_hidden_size,

View File

@@ -77,6 +77,7 @@ class WuerstchenPriorPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
def dummy_text_encoder(self):
torch.manual_seed(0)
config = CLIPTextConfig(
attn_implementation="eager",
bos_token_id=0,
eos_token_id=2,
hidden_size=self.text_embedder_hidden_size,