Spaces:

nkanungo
/

tinygpt

Runtime error

nkanungo commited on Oct 27, 2023

Commit

f77e58d

1 Parent(s): decf5fd

Update nano_gpt_inferencing.py

Files changed (1) hide show

nano_gpt_inferencing.py CHANGED Viewed

@@ -187,7 +187,7 @@ class BigramLanguageModel(nn.Module):
         return idx
-def generate_paragraph(initial_text,max_token):
     model = BigramLanguageModel()
     model.load_state_dict(torch.load('nano_gpt_model.pth'))
     final_model = model.to(device)

         return idx
+def generate_paragraph(initial_text,max_token=50):
     model = BigramLanguageModel()
     model.load_state_dict(torch.load('nano_gpt_model.pth'))
     final_model = model.to(device)