rovi27 commited on
Commit
072e662
·
verified ·
1 Parent(s): 4b87c3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ sft_model = "somosnlp/RecetasDeLaAbuela5k_gemma-2b-bnb-4bit"
13
  base_model_name = "unsloth/gemma-2b-bnb-4bit"
14
  #base_model_name = "unsloth/gemma-2b-it-bnb-4bit"
15
 
16
- max_seq_length=400
17
  base_model = AutoModelForCausalLM.from_pretrained(base_model_name,return_dict=True,device_map="auto", torch_dtype=torch.float16,)
18
  tokenizer = AutoTokenizer.from_pretrained(base_model_name, max_length = max_seq_length)
19
  ft_model = PeftModel.from_pretrained(base_model, sft_model)
 
13
  base_model_name = "unsloth/gemma-2b-bnb-4bit"
14
  #base_model_name = "unsloth/gemma-2b-it-bnb-4bit"
15
 
16
+ max_seq_length=200
17
  base_model = AutoModelForCausalLM.from_pretrained(base_model_name,return_dict=True,device_map="auto", torch_dtype=torch.float16,)
18
  tokenizer = AutoTokenizer.from_pretrained(base_model_name, max_length = max_seq_length)
19
  ft_model = PeftModel.from_pretrained(base_model, sft_model)