slobers
/

sos01

slobers commited on Dec 13, 2024

Commit

2c86ed3

verified ·

1 Parent(s): 28e146e

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py CHANGED Viewed

@@ -793,7 +793,7 @@ def load_pipeline() -> Pipeline:
     text_encoder_2 = T5EncoderModel.from_pretrained(CHECKPOINT, revision=REVISION, subfolder="text_encoder_2", local_files_only=True, torch_dtype=torch.bfloat16,)
-    path1 = os.path.join(HF_HUB_CACHE, "models--aifeifei798--taef1/snapshots/319774e8256cc42d98b3c2790ac2c5edc8a73389")
     vae = AutoencoderTiny.from_pretrained(path1, torch_dtype=torch.bfloat16,)

     text_encoder_2 = T5EncoderModel.from_pretrained(CHECKPOINT, revision=REVISION, subfolder="text_encoder_2", local_files_only=True, torch_dtype=torch.bfloat16,)
+    path1 = os.path.join(HF_HUB_CACHE, "models--madebyollin--taef1/snapshots/5463ee684fd9131a724bea777a2f50d89b0b6b24")
     vae = AutoencoderTiny.from_pretrained(path1, torch_dtype=torch.bfloat16,)