Hugging Face's logo

Spaces:
HimankJ
/
GPT2_CustomTrained
Runtime error

App Files Community
HimankJ commited on
Commit
cd181af
·
verified ·
1 Parent(s): 3eaa4f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ max_length = 500
21
  enc = tiktoken.get_encoding('gpt2')
22
 
23
  # CHANGES IN CURRENT CODE
24
- ckpt_path = os.path.join(modelpath, 'GPT2_TinyShakespeare2.pt')
25
  print(ckpt_path)
26
  checkpoint = torch.load(ckpt_path, map_location=device)
27
  gptconf = GPTConfig(**checkpoint['model_args'])
 
21
  enc = tiktoken.get_encoding('gpt2')
22
 
23
  # CHANGES IN CURRENT CODE
24
+ ckpt_path = os.path.join(modelpath, 'GPT2_TinyShakespeare1.pt')
25
  print(ckpt_path)
26
  checkpoint = torch.load(ckpt_path, map_location=device)
27
  gptconf = GPTConfig(**checkpoint['model_args'])