manbeast3b
/

deserve_edge4_iter1

Model card Files Files and versions

manbeast3b commited on Dec 17, 2024

Commit

9aa6b34

·

verified ·

1 Parent(s): 6e0b3f7

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -594,7 +594,7 @@ def load_pipeline() -> Pipeline:
         "city96/t5-v1_1-xxl-encoder-bf16", revision = "1b9c856aadb864af93c1dcdc226c2774fa67bc86", torch_dtype=torch.bfloat16
     ).to(memory_format=torch.channels_last)
-    vae = AutoencoderTiny.from_pretrained("RobertML/FLUX.1-schnell-vae_int8", revision="2e4fdf1337ed12ed202e23e6f25269aa7d6d3f2a", torch_dtype=dtype)
     path = os.path.join(HF_HUB_CACHE, "models--RobertML--FLUX.1-schnell-int8wo/snapshots/307e0777d92df966a3c0f99f31a6ee8957a9857a")
     generator = torch.Generator(device=device)

         "city96/t5-v1_1-xxl-encoder-bf16", revision = "1b9c856aadb864af93c1dcdc226c2774fa67bc86", torch_dtype=torch.bfloat16
     ).to(memory_format=torch.channels_last)
+    vae = AutoencoderKL.from_pretrained("RobertML/FLUX.1-schnell-vae_int8", revision="2e4fdf1337ed12ed202e23e6f25269aa7d6d3f2a", torch_dtype=dtype)
     path = os.path.join(HF_HUB_CACHE, "models--RobertML--FLUX.1-schnell-int8wo/snapshots/307e0777d92df966a3c0f99f31a6ee8957a9857a")
     generator = torch.Generator(device=device)