ZhengPeng7 commited on
Commit
6f22264
1 Parent(s): 40a7dcf

Ensemble BiRefNet codes to HF to load model with weights in one line.

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -53,7 +53,7 @@ import torch
53
  from torchvision import transforms
54
  from models.birefnet import BiRefNet
55
 
56
- # birefnet = ... -- BiRefNet has been loaded with codes above, either way.
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()