lukisko commited on
Commit
ef2b0a0
·
verified ·
1 Parent(s): b2d4c7d

go back to 1.7B

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from huggingface_hub import login
8
  login(os.getenv('HF_LOGIN'))
9
 
10
  token_step_size = 20
11
- model_id = "utter-project/EuroLLM-9B-Instruct"
12
  tokenizer = AutoTokenizer.from_pretrained(model_id, torch_dtype=torch.bfloat16)
13
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
14
  model.generation_config.pad_token_id = tokenizer.pad_token_id
 
8
  login(os.getenv('HF_LOGIN'))
9
 
10
  token_step_size = 20
11
+ model_id = "utter-project/EuroLLM-1.7B-Instruct"
12
  tokenizer = AutoTokenizer.from_pretrained(model_id, torch_dtype=torch.bfloat16)
13
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
14
  model.generation_config.pad_token_id = tokenizer.pad_token_id