Update CLIP.py

CLIP.py

@@ -11,7 +11,7 @@ def load(tokenizer_path = "tokenizer", text_encoder_path = "text_encoder"):
     config_path = f"./{text_encoder_path}/config.json"
     
     # Load tokenizer
-    tokenizer = CLIPTokenizer.from_pretrained(tokenizer_path)
+    clip_tokenizer = CLIPTokenizer.from_pretrained(tokenizer_path)
     
     # Load CLIPTextModelWithProjection from the config file and safetensor
     clip_model = CLIPTextModelWithProjection.from_pretrained(
@@ -25,4 +25,12 @@ def load(tokenizer_path = "tokenizer", text_encoder_path = "text_encoder"):
     clip_model.load_state_dict(state_dict)
     clip_model = clip_model.to(device)
     
-    return clip_model, tokenizer
+    return clip_model, clip_tokenizer
+
+def load_vae(vae_path='vae'):
+    return AutoencoderKL.from_pretrained(vae_pathj)
+
+# Example function for processing prompts
+def encode_prompt(prompt,tokenizer,clip_model):
+    inputs = tokenizer(prompt, return_tensors="pt")
+    return clip_model(**inputs).last_hidden_state