K00B404 commited on
Commit
cfce18c
·
verified ·
1 Parent(s): 5f1eb4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,16 +23,16 @@ ipadapter_path = hf_hub_download(repo_id="InstantX/FLUX.1-dev-IP-Adapter", filen
23
  transformer = FluxTransformer2DModel.from_pretrained(
24
  "black-forest-labs/FLUX.1-dev",
25
  subfolder="transformer",
26
- torch_dtype=torch.bfloat16
27
  )
28
 
29
  pipe = FluxPipeline.from_pretrained(
30
  "black-forest-labs/FLUX.1-dev",
31
  transformer=transformer,
32
- torch_dtype=torch.bfloat16
33
  )
34
 
35
- ip_model = IPAdapter(pipe, image_encoder_path, ipadapter_path, device="cuda", num_tokens=128)
36
 
37
  def resize_img(image, max_size=1024):
38
  width, height = image.size
 
23
  transformer = FluxTransformer2DModel.from_pretrained(
24
  "black-forest-labs/FLUX.1-dev",
25
  subfolder="transformer",
26
+
27
  )
28
 
29
  pipe = FluxPipeline.from_pretrained(
30
  "black-forest-labs/FLUX.1-dev",
31
  transformer=transformer,
32
+
33
  )
34
 
35
+ ip_model = IPAdapter(pipe, image_encoder_path, ipadapter_path, device="cpu", num_tokens=128)
36
 
37
  def resize_img(image, max_size=1024):
38
  width, height = image.size