Update app.py

app.py

@@ -53,6 +53,8 @@ clip_model = CLIPModel.from_pretrained(model_id, torch_dtype=torch.bfloat16, con
 clip_processor = CLIPProcessor.from_pretrained(model_id, padding="max_length", max_length=maxtokens, ignore_mismatched_sizes=True, return_tensors="pt", truncation=True)
 config.text_config.max_position_embeddings = 77
 
+_optional_components = [transformer, scheduler, vae, text_encoder, text_encoder_2, text_encoder_3, tokenizer, tokenizer_2, tokenizer_3]
+
 pipe.tokenizer = clip_processor.tokenizer
 pipe.text_encoder = clip_model.text_model
 pipe.tokenizer_max_length = maxtokens