manbeast3b
/

bench_try10

Model card Files Files and versions

manbeast3b commited on Dec 25, 2024

Commit

5e48d3b

·

verified ·

1 Parent(s): 560c690

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +2 -2

src/pipeline.py CHANGED Viewed

@@ -167,8 +167,8 @@ def load_pipeline() -> Pipeline:
     quantize_(vae, int8_weight_only())
-    text_encoder_2 = T5EncoderModel.from_pretrained("edgetensor/edgetensor-t5-v1_1-xxl-encoder-bf16", revision = "bfed5213335c2ead9c9b5aff657680db420a7c7d", torch_dtype=torch.bfloat16).to(memory_format=torch.channels_last)
-    path = os.path.join(HF_HUB_CACHE, "models--edgetensor--edgetensor-FLUX.1-schnell-int8wo/snapshots/6b4c594d4c510da13a40f8c3b483789eb82d36df")
     transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16, use_safetensors=False).to(memory_format=torch.channels_last)
     pipeline = DiffusionPipeline.from_pretrained(ids, revision=Revision, transformer=transformer, text_encoder_2=text_encoder_2, torch_dtype=torch.bfloat16,)

     quantize_(vae, int8_weight_only())
+    text_encoder_2 = T5EncoderModel.from_pretrained("city96/t5-v1_1-xxl-encoder-bf16", revision = "1b9c856aadb864af93c1dcdc226c2774fa67bc86", torch_dtype=torch.bfloat16).to(memory_format=torch.channels_last)
+    path = os.path.join(HF_HUB_CACHE, "models--RobertML--FLUX.1-schnell-int8wo/snapshots/307e0777d92df966a3c0f99f31a6ee8957a9857a")
     transformer = FluxTransformer2DModel.from_pretrained(path, torch_dtype=torch.bfloat16, use_safetensors=False).to(memory_format=torch.channels_last)
     pipeline = DiffusionPipeline.from_pretrained(ids, revision=Revision, transformer=transformer, text_encoder_2=text_encoder_2, torch_dtype=torch.bfloat16,)