Spaces:
Running
Running
Update CLIP.py
Browse files
CLIP.py
CHANGED
@@ -30,7 +30,7 @@ def load(tokenizer_path = "tokenizer", text_encoder_path = "text_encoder"):
|
|
30 |
return clip_model, clip_tokenizer
|
31 |
|
32 |
def load_vae(vae_path='vae'):
|
33 |
-
return AutoencoderKL.from_pretrained(
|
34 |
|
35 |
# Example function for processing prompts
|
36 |
def encode_prompt(prompt,tokenizer,clip_model):
|
|
|
30 |
return clip_model, clip_tokenizer
|
31 |
|
32 |
def load_vae(vae_path='vae'):
|
33 |
+
return AutoencoderKL.from_pretrained(vae_path)
|
34 |
|
35 |
# Example function for processing prompts
|
36 |
def encode_prompt(prompt,tokenizer,clip_model):
|