ZhengPeng7
commited on
Commit
•
6f22264
1
Parent(s):
40a7dcf
Ensemble BiRefNet codes to HF to load model with weights in one line.
Browse files
README.md
CHANGED
@@ -53,7 +53,7 @@ import torch
|
|
53 |
from torchvision import transforms
|
54 |
from models.birefnet import BiRefNet
|
55 |
|
56 |
-
|
57 |
torch.set_float32_matmul_precision(['high', 'highest'][0])
|
58 |
birefnet.to('cuda')
|
59 |
birefnet.eval()
|
|
|
53 |
from torchvision import transforms
|
54 |
from models.birefnet import BiRefNet
|
55 |
|
56 |
+
birefnet = ... # -- BiRefNet should be loaded with codes above, either way.
|
57 |
torch.set_float32_matmul_precision(['high', 'highest'][0])
|
58 |
birefnet.to('cuda')
|
59 |
birefnet.eval()
|