Spaces:

not-lain
/

background-removal

Running on Zero

not-lain commited on Aug 8, 2024

Commit

f3d3ce9

1 Parent(s): c368dca

🌘w🌖

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,9 +6,9 @@ from transformers import AutoModelForImageSegmentation
 import torch
 from torchvision import transforms
-# torch.set_float32_matmul_precision(['high', 'highest'][0])
-birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True,torch_dtype=torch.float16)
 birefnet.to("cuda")
 transform_image = transforms.Compose([
     transforms.Resize((1024, 1024)),

 import torch
 from torchvision import transforms
+torch.set_float32_matmul_precision(['high', 'highest'][0])
+birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)
 birefnet.to("cuda")
 transform_image = transforms.Compose([
     transforms.Resize((1024, 1024)),