BounharAbdelaziz commited on
Commit
fb2bae2
·
verified ·
1 Parent(s): 309169c

added access token

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -4,15 +4,17 @@ import os
4
  import spaces
5
  import torch
6
 
 
 
7
 
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
  print(f'[INFO] Using device: {device}')
10
 
11
  # Load the pretrained model and tokenizer
12
- MODEL_NAME = "BounharAbdelaziz/Al-Atlas-LLM-0.5B" # "atlasia/Al-Atlas-LLM-mid-training" # "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
13
 
14
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME) # , token=token
15
- model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to(device)
16
 
17
  # Fix tokenizer padding
18
  if tokenizer.pad_token is None:
 
4
  import spaces
5
  import torch
6
 
7
+ TOKEN = os.environ['TOKEN']
8
+
9
 
10
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
11
  print(f'[INFO] Using device: {device}')
12
 
13
  # Load the pretrained model and tokenizer
14
+ MODEL_NAME = "BounharAbdelaziz/Al-Atlas-LLM-0.5B"
15
 
16
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=TOKEN)
17
+ model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=TOKEN).to(device)
18
 
19
  # Fix tokenizer padding
20
  if tokenizer.pad_token is None: