DOFOFFICIAL
commited on
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")
|