Commit
·
79a6660
1
Parent(s):
45e89f1
Move all BiRefNet github codes to the first level directory.
Browse files- BiRefNet_pipe.py +0 -1
- __init__.py +0 -0
BiRefNet_pipe.py
CHANGED
|
@@ -7,4 +7,3 @@ class BiRefNetPipe(Pipeline):
|
|
| 7 |
Pipeline.__init__(self, **kwargs)
|
| 8 |
self.model.to(['cpu', 0][torch.cuda.is_available()])
|
| 9 |
self.model.eval()
|
| 10 |
-
|
|
|
|
| 7 |
Pipeline.__init__(self, **kwargs)
|
| 8 |
self.model.to(['cpu', 0][torch.cuda.is_available()])
|
| 9 |
self.model.eval()
|
|
|
__init__.py
ADDED
|
File without changes
|