Update app.py

app.py

@@ -109,12 +109,10 @@ def load_and_prepare_model():
     vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False, low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True) #.to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
     pipe = StableDiffusionXLPipeline.from_pretrained(
         'ford442/RealVisXL_V5.0_BF16',
-       #'ford442/Juggernaut-XI-v11-fp32',
       #   'SG161222/RealVisXL_V5.0',
         #'John6666/uber-realistic-porn-merge-xl-urpmxl-v3-sdxl',
         #torch_dtype=torch.bfloat16,
         add_watermarker=False,
-       # custom_pipeline="lpw_stable_diffusion_xl",
         #use_safetensors=True,
         token=HF_TOKEN,
         text_encoder=None,
@@ -142,7 +140,8 @@ def load_and_prepare_model():
     #pipe.vae.do_resize=False
     #pipe.vae.do_rescale=False
     #pipe.vae.do_convert_rgb=True
-    #pipe.vae.vae_scale_factor=8    #pipe.unet.set_default_attn_processor()
+    #pipe.vae.vae_scale_factor=8
+    #pipe.unet.set_default_attn_processor()
     pipe.vae.set_default_attn_processor()
     print(f'Pipeline: ')
     #print(f'_optional_components: {pipe._optional_components}')
@@ -163,9 +162,7 @@ checkpoint = "microsoft/Phi-3.5-mini-instruct"
 captioner = pipeline(model="ydshieh/vit-gpt2-coco-en",device='cuda', task="image-to-text")
 captioner_2 = pipeline(model="Salesforce/blip-image-captioning-base",device='cuda', task="image-to-text")
 captioner_3 = pipeline(model="Salesforce/blip-image-captioning-large",device='cuda', task="image-to-text")
-#model5 = InstructBlipForConditionalGeneration.from_pretrained("Salesforce/instructblip-vicuna-7b").to(torch.bfloat16).to('cuda')
 model5 = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b-coco").to('cuda')
-#processor5 = InstructBlipProcessor.from_pretrained("Salesforce/instructblip-vicuna-7b").to(torch.bfloat16).to('cuda')
 processor5 = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b-coco")
 txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
 txt_tokenizer.tokenizer_legacy=False