Return to AutoPipelineForText2Image

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import numpy as np
 import random
 import os
-from diffusers import AutoPipelineForImage2Image
+from diffusers import AutoPipelineForText2Image
 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 = AutoPipelineForImage2Image.from_pretrained(
+pipeline = AutoPipelineForText2Image.from_pretrained(
     'black-forest-labs/FLUX.1-schnell', torch_dtype=torch.bfloat16).to('cuda')
 pipeline.load_lora_weights('Roomie/xavyy', weight_name='xavyy.safetensors')
 
@@ -32,7 +32,7 @@ def infer(prompt, refer_image, height, width):
         # num_inference_steps=num_inference_steps,
         width=width,
         height=height,
-        image=refer_image
+        # image=refer_image
         # generator=generator
     ).images[0]