Use torch save.
Browse files- app.py +3 -2
- model_v4t.pkl +3 -0
app.py
CHANGED
@@ -108,8 +108,9 @@ class Model(nn.Module):
|
|
108 |
m = Model()
|
109 |
m.to(device)
|
110 |
|
111 |
-
with open("model_v4.pkl","rb") as f:
|
112 |
-
m=pickle.load(f)
|
|
|
113 |
|
114 |
top_k = 20
|
115 |
def generate(s, num = 60):
|
|
|
108 |
m = Model()
|
109 |
m.to(device)
|
110 |
|
111 |
+
#with open("model_v4.pkl","rb") as f:
|
112 |
+
#m=pickle.load(f)
|
113 |
+
m=torch.load("model_v4t.pkl",map_location=torch.device('cpu'))
|
114 |
|
115 |
top_k = 20
|
116 |
def generate(s, num = 60):
|
model_v4t.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e56c8c0d3d1fb4c25547214ed8a32f2e22fb71a1273cbd6fba4ab2661e2a319
|
3 |
+
size 4132106
|