Hugging Face's logo

slobers
/
sos02

Model card Files
xet
 Community
slobers commited on
Commit
7c27ff4
·
verified ·
1 Parent(s): 64c779b

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -19,7 +19,7 @@ REVISION = "741f7c3ce8b383c54771c7003378a50191e9efe9"
19
  def load_pipeline() -> Pipeline:
20
  text_encoder = CLIPTextModel.from_pretrained(CHECKPOINT, revision=REVISION, subfolder="text_encoder", local_files_only=True, torch_dtype=torch.bfloat16,)
21
 
22
- patht2 = os.path.join(HF_HUB_CACHE, "models--city96--t5-v1_1-xxl-encoder-bf16/snapshots/1b9c856aadb864af93c1dcdc226c2774fa67bc86")
23
 
24
  text_encoder_2 = T5EncoderModel.from_pretrained(path2, torch_dtype=torch.bfloat16,)
25
 
 
19
  def load_pipeline() -> Pipeline:
20
  text_encoder = CLIPTextModel.from_pretrained(CHECKPOINT, revision=REVISION, subfolder="text_encoder", local_files_only=True, torch_dtype=torch.bfloat16,)
21
 
22
+ path2 = os.path.join(HF_HUB_CACHE, "models--city96--t5-v1_1-xxl-encoder-bf16/snapshots/1b9c856aadb864af93c1dcdc226c2774fa67bc86")
23
 
24
  text_encoder_2 = T5EncoderModel.from_pretrained(path2, torch_dtype=torch.bfloat16,)
25