svjack commited on
Commit
e20ee6f
·
verified ·
1 Parent(s): a26a6f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -31,7 +31,7 @@ ghibli_style_lora_path = hf_hub_download(repo_id="InstantX/FLUX.1-dev-LoRA-Ghibl
31
 
32
  # init InstantCharacter pipeline
33
  pipe = InstantCharacterFluxPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
34
- pipe.to(device)
35
 
36
  # load InstantCharacter
37
  pipe.init_adapter(
@@ -40,6 +40,8 @@ pipe.init_adapter(
40
  subject_ipadapter_cfg=dict(subject_ip_adapter_path=ip_adapter_path, nb_token=1024),
41
  )
42
 
 
 
43
  # load matting model
44
  birefnet = AutoModelForImageSegmentation.from_pretrained(birefnet_path, trust_remote_code=True)
45
  birefnet.to('cuda')
@@ -302,4 +304,4 @@ with block:
302
 
303
  gr.Markdown(article)
304
 
305
- block.launch()
 
31
 
32
  # init InstantCharacter pipeline
33
  pipe = InstantCharacterFluxPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
34
+ #pipe.to(device)
35
 
36
  # load InstantCharacter
37
  pipe.init_adapter(
 
40
  subject_ipadapter_cfg=dict(subject_ip_adapter_path=ip_adapter_path, nb_token=1024),
41
  )
42
 
43
+ pipe.enable_sequential_cpu_offload()
44
+
45
  # load matting model
46
  birefnet = AutoModelForImageSegmentation.from_pretrained(birefnet_path, trust_remote_code=True)
47
  birefnet.to('cuda')
 
304
 
305
  gr.Markdown(article)
306
 
307
+ block.launch(share = True)