Ernestasio commited on
Commit
f6e8142
·
1 Parent(s): 17854c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ device = "cuda"
7
  model_name = 'mistralai/Mistral-7B-Instruct-v0.1'
8
 
9
  # Cargar el modelo preentrenado y el tokenizador asociado
10
- model = AutoModelForCausalLM.from_pretrained(model_name)
11
 
12
 
13
  # Cargar el tokenizador asociado
 
7
  model_name = 'mistralai/Mistral-7B-Instruct-v0.1'
8
 
9
  # Cargar el modelo preentrenado y el tokenizador asociado
10
+ model = AutoModelForCausalLM.from_pretrained(model_name, load_in_8bit=True,)
11
 
12
 
13
  # Cargar el tokenizador asociado