valhalla commited on
Commit
bc5c316
1 Parent(s): 85c60b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ import torch.nn.functional as F
5
 
6
  from transformers import XGLMTokenizer, XGLMForCausalLM
7
 
8
- tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-1.7B")
9
- model = XGLMForCausalLM.from_pretrained("facebook/xglm-1.7B")
10
 
11
  data_samples = {
12
  'en': [
5
 
6
  from transformers import XGLMTokenizer, XGLMForCausalLM
7
 
8
+ tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-2.9B")
9
+ model = XGLMForCausalLM.from_pretrained("facebook/xglm-2.9B")
10
 
11
  data_samples = {
12
  'en': [