Spaces:
Paused
Paused
uuu
Browse files
app.py
CHANGED
@@ -27,8 +27,6 @@ from huggingface_hub import login
|
|
27 |
token = os.getenv("HF_TOKEN")
|
28 |
login(token=token)
|
29 |
|
30 |
-
# Model and Pipeline Setup
|
31 |
-
|
32 |
model_path = 'stabilityai/stable-diffusion-3.5-large'
|
33 |
ip_adapter_path = './ip-adapter.bin'
|
34 |
image_encoder_path = "google/siglip-so400m-patch14-384"
|
@@ -48,21 +46,6 @@ pipe.init_ipadapter(
|
|
48 |
)
|
49 |
|
50 |
|
51 |
-
# Load transformer and pipeline
|
52 |
-
transformer = SD3Transformer2DModel.from_pretrained(
|
53 |
-
model_path, subfolder="transformer", torch_dtype=torch.bfloat16
|
54 |
-
)
|
55 |
-
pipe = StableDiffusion3Pipeline.from_pretrained(
|
56 |
-
model_path, transformer=transformer, torch_dtype=torch.bfloat16
|
57 |
-
).to("cuda")
|
58 |
-
|
59 |
-
# Initialize IP Adapter
|
60 |
-
pipe.init_ipadapter(
|
61 |
-
ip_adapter_path=ip_adapter_path,
|
62 |
-
image_encoder_path=image_encoder_path,
|
63 |
-
nb_token=64,
|
64 |
-
)
|
65 |
-
|
66 |
@spaces.GPU
|
67 |
def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
|
68 |
|
@@ -76,9 +59,7 @@ def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
|
|
76 |
guidance_scale=guidance_scale,
|
77 |
generator=torch.Generator("cuda").manual_seed(42),
|
78 |
clip_image=ref_img.convert('RGB'),
|
79 |
-
ipadapter_scale=ipadapter_scale
|
80 |
-
).images[0]
|
81 |
-
|
82 |
return image
|
83 |
|
84 |
# Create Gradio interface
|
|
|
27 |
token = os.getenv("HF_TOKEN")
|
28 |
login(token=token)
|
29 |
|
|
|
|
|
30 |
model_path = 'stabilityai/stable-diffusion-3.5-large'
|
31 |
ip_adapter_path = './ip-adapter.bin'
|
32 |
image_encoder_path = "google/siglip-so400m-patch14-384"
|
|
|
46 |
)
|
47 |
|
48 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
49 |
@spaces.GPU
|
50 |
def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
|
51 |
|
|
|
59 |
guidance_scale=guidance_scale,
|
60 |
generator=torch.Generator("cuda").manual_seed(42),
|
61 |
clip_image=ref_img.convert('RGB'),
|
62 |
+
ipadapter_scale=ipadapter_scale).images[0]
|
|
|
|
|
63 |
return image
|
64 |
|
65 |
# Create Gradio interface
|