Hugging Face's logo

Spaces:
Kc-12
/
TinyStories_Transformer
Sleeping

App Files Community
3
Kc-12 commited on
Commit
1d3e91d
·
1 Parent(s): 429022a

Upload better_transformer.py

Browse files
Files changed (1) hide show
  1. better_transformer.py +0 -1
better_transformer.py CHANGED
@@ -9,7 +9,6 @@ import torch
9
  import torch.nn as nn
10
  import torch.nn.functional as F
11
  from transformers import AutoTokenizer
12
- from torchinfo import summary
13
 
14
  MODEL_FILE = r'bt_8_LAYERs_100_DATA_PCT_768_EMBD_DIM_epoch_10.pt' ##place model file in same directory as app.py
15
  torch.set_default_device(torch.device("cuda"))
 
9
  import torch.nn as nn
10
  import torch.nn.functional as F
11
  from transformers import AutoTokenizer
 
12
 
13
  MODEL_FILE = r'bt_8_LAYERs_100_DATA_PCT_768_EMBD_DIM_epoch_10.pt' ##place model file in same directory as app.py
14
  torch.set_default_device(torch.device("cuda"))