Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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)
|