AutoPipelineForImage2Image test

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import numpy as np
 import random
 import os
-from diffusers import FluxImg2ImgPipeline
+from diffusers import AutoPipelineForImage2Image
 import torch
 from huggingface_hub import snapshot_download
 
@@ -12,7 +12,7 @@ SPACE_PASSWORD = 'KR_4dmin'
 
 snapshot_download(repo_id="Roomie/xavyy", cache_dir='./')
 
-pipeline = FluxImg2ImgPipeline.from_pretrained(
+pipeline = AutoPipelineForImage2Image.from_pretrained(
     'black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cuda')
 pipeline.load_lora_weights('Roomie/xavyy', weight_name='xavyy.safetensors')