ibrim commited on
Commit
5b5b19a
1 Parent(s): 7dcfe75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import tiktoken
6
  from model import GPTConfig, GPT
7
  import gradio as gr
8
 
9
- def sample_from_trained_model(start="\n", init_from='resume', out_dir='ibrim/NanoGPT/out-shakespeare-char', num_samples=1,
10
  max_new_tokens=500, temperature=0.8, top_k=200, seed=1337, device='cpu', compile=False):
11
  # Set the dtype
12
  dtype = 'bfloat16' if torch.cuda.is_available() and torch.cuda.is_bf16_supported() else 'float16'
 
6
  from model import GPTConfig, GPT
7
  import gradio as gr
8
 
9
+ def sample_from_trained_model(start="\n", init_from='resume', out_dir='out-shakespeare-char', num_samples=1,
10
  max_new_tokens=500, temperature=0.8, top_k=200, seed=1337, device='cpu', compile=False):
11
  # Set the dtype
12
  dtype = 'bfloat16' if torch.cuda.is_available() and torch.cuda.is_bf16_supported() else 'float16'