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()
|