Spaces:

somosnlp
/

Colombia_RAC

Runtime error

NickyNicky commited on Apr 24, 2024

Commit

67fc3b6

verified ·

1 Parent(s): 0f3838e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 # from google.colab import userdata
 import os
-model_id = "somosnlp/gemma-1.1-2b-it_ColombiaRAC_FullyCurated_format_chatML_V1"
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,
     bnb_4bit_quant_type="nf4",

 # from google.colab import userdata
 import os
+model_id = "somosnlp/GemmaColRAC-AeroExpert"
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,
     bnb_4bit_quant_type="nf4",