Commit ·
2671a55
1
Parent(s): b812c98
Update usage/interfere_cpu.py
Browse files- usage/interfere_cpu.py +4 -1
usage/interfere_cpu.py
CHANGED
|
@@ -78,7 +78,10 @@ def predictmain(model, filepath):
|
|
| 78 |
|
| 79 |
if __name__ == '__main__':
|
| 80 |
# local usage
|
| 81 |
-
model = modelload("
|
|
|
|
|
|
|
|
|
|
| 82 |
|
| 83 |
# use your picture to interfere
|
| 84 |
cmpgraph_64x64("path.png", "path(1).png")
|
|
|
|
| 78 |
|
| 79 |
if __name__ == '__main__':
|
| 80 |
# local usage
|
| 81 |
+
model = modelload("pytorch_model.bin")
|
| 82 |
+
|
| 83 |
+
# or uses safetensors
|
| 84 |
+
# model = torch.load("pytorch_model.safetensors")
|
| 85 |
|
| 86 |
# use your picture to interfere
|
| 87 |
cmpgraph_64x64("path.png", "path(1).png")
|