ehristoforu commited on
Commit
5d47c47
·
verified ·
1 Parent(s): d8e7c34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -29,9 +29,11 @@ model = AutoModelForCausalLM.from_pretrained(
29
  torch_dtype=torch.float16,
30
  device_map="auto"
31
  )
32
- tokenizer = AutoTokenizer.from_pretrained(model_name)
33
 
34
  model.save_pretrained("./falconthink")
 
 
35
 
36
  from huggingface_hub import HfApi
37
 
 
29
  torch_dtype=torch.float16,
30
  device_map="auto"
31
  )
32
+ tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-10B-Instruct")
33
 
34
  model.save_pretrained("./falconthink")
35
+ tokenizer.save_pretrained("./falconthink")
36
+
37
 
38
  from huggingface_hub import HfApi
39