gagan001 commited on
Commit
c569724
·
1 Parent(s): 4fbeca8

Added cpu model

Browse files
app.py CHANGED
@@ -4,7 +4,10 @@ from my_gpt import my_gpt
4
  from tokenizer.tokenizer import BPE
5
 
6
  ##Load model
7
- model = my_gpt.load_pretrained("model/model_1000_.bin")
 
 
 
8
  tokenizer = BPE()
9
 
10
 
 
4
  from tokenizer.tokenizer import BPE
5
 
6
  ##Load model
7
+ model = my_gpt.load_pretrained("model/model_1000_cpu.bin")
8
+ # model.to(torch.device("cpu"))
9
+ # model.save_pretrained("model/model_1000_cpu.bin")
10
+ # exit()
11
  tokenizer = BPE()
12
 
13
 
model/{model_1000_.bin → model_1000_cpu.bin} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:566dd60c0869b306eda84fd1d64dbacce02b256492da9ea626d227b046125bd2
3
- size 56950645
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f88eff25b6947e11a832f96e2bc914c6818989045539c327438c3e490b184cc9
3
+ size 56951293