mirror of
https://github.com/huggingface/diffusers.git
synced 2025-12-10 06:24:19 +08:00
Compare commits
1 Commits
torchao-of
...
tests-clip
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b6e47d82dd |
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user