Spaces:
Running
on
A100
Running
on
A100
Update app.py
Browse files
app.py
CHANGED
@@ -124,6 +124,8 @@ models_rbm = core.Models(
|
|
124 |
)
|
125 |
models_rbm.generator.eval().requires_grad_(False)
|
126 |
|
|
|
|
|
127 |
def infer(ref_style_file, style_description, caption):
|
128 |
global models_rbm, models_b, device
|
129 |
if low_vram:
|
@@ -249,7 +251,7 @@ def infer_compo(style_description, ref_style_file, caption, ref_sub_file):
|
|
249 |
## SAM Mask for sub
|
250 |
use_sam_mask = False
|
251 |
x0_preview = models_rbm.previewer(x0_forward)
|
252 |
-
|
253 |
sam_mask, boxes, phrases, logits = sam_model.predict(transform(x0_preview[0]), sam_prompt)
|
254 |
sam_mask = sam_mask.detach().unsqueeze(dim=0).to(device)
|
255 |
|
|
|
124 |
)
|
125 |
models_rbm.generator.eval().requires_grad_(False)
|
126 |
|
127 |
+
sam_model = LangSAM()
|
128 |
+
|
129 |
def infer(ref_style_file, style_description, caption):
|
130 |
global models_rbm, models_b, device
|
131 |
if low_vram:
|
|
|
251 |
## SAM Mask for sub
|
252 |
use_sam_mask = False
|
253 |
x0_preview = models_rbm.previewer(x0_forward)
|
254 |
+
|
255 |
sam_mask, boxes, phrases, logits = sam_model.predict(transform(x0_preview[0]), sam_prompt)
|
256 |
sam_mask = sam_mask.detach().unsqueeze(dim=0).to(device)
|
257 |
|