Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -53,8 +53,6 @@ clip_model = CLIPModel.from_pretrained(model_id, torch_dtype=torch.bfloat16, con
|
|
53 |
clip_processor = CLIPProcessor.from_pretrained(model_id, padding="max_length", max_length=maxtokens, ignore_mismatched_sizes=True, return_tensors="pt", truncation=True)
|
54 |
config.text_config.max_position_embeddings = 77
|
55 |
|
56 |
-
_optional_components = [transformer, scheduler, vae, text_encoder, text_encoder_2, text_encoder_3, tokenizer, tokenizer_2, tokenizer_3]
|
57 |
-
|
58 |
pipe.tokenizer = clip_processor.tokenizer
|
59 |
pipe.text_encoder = clip_model.text_model
|
60 |
pipe.tokenizer_max_length = maxtokens
|
|
|
53 |
clip_processor = CLIPProcessor.from_pretrained(model_id, padding="max_length", max_length=maxtokens, ignore_mismatched_sizes=True, return_tensors="pt", truncation=True)
|
54 |
config.text_config.max_position_embeddings = 77
|
55 |
|
|
|
|
|
56 |
pipe.tokenizer = clip_processor.tokenizer
|
57 |
pipe.text_encoder = clip_model.text_model
|
58 |
pipe.tokenizer_max_length = maxtokens
|