amos1088 commited on
Commit
5378962
·
1 Parent(s): 13a9d06
Files changed (1) hide show
  1. app.py +1 -20
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