AlekseyCalvin commited on
Commit
2f4b4f6
·
verified ·
1 Parent(s): 11e846d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
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