manbeast3b
/

deserve_edge4_a3

Model card Files Files and versions

manbeast3b commited on Dec 19, 2024

Commit

616d50a

·

verified ·

1 Parent(s): e03447b

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -599,7 +599,7 @@ def load_pipeline() -> Pipeline:
     ).to(memory_format=torch.channels_last)
     text_encoder = CLIPTextModel.from_pretrained(
-        os.path.join(HF_HUB_CACHE, "models--manbeast3b--FLUX.1-schnell-te1/snapshots/05ac3e466d6b42b7794859560d875b25f6df5daf"), subfolder="text_encoder", torch_dtype=torch.bfloat16
     ).to(memory_format=torch.channels_last)
     vae = AutoencoderTiny.from_pretrained("RobertML/FLUX.1-schnell-vae_e3m2", revision="da0d2cd7815792fb40d084dbd8ed32b63f153d8d", torch_dtype=dtype)

     ).to(memory_format=torch.channels_last)
     text_encoder = CLIPTextModel.from_pretrained(
+        os.path.join(HF_HUB_CACHE, "models--manbeast3b--FLUX.1-schnell-te1/snapshots/05ac3e466d6b42b7794859560d875b25f6df5daf"), torch_dtype=torch.bfloat16
     ).to(memory_format=torch.channels_last)
     vae = AutoencoderTiny.from_pretrained("RobertML/FLUX.1-schnell-vae_e3m2", revision="da0d2cd7815792fb40d084dbd8ed32b63f153d8d", torch_dtype=dtype)