jordigonzm commited on
Commit
50c763f
1 Parent(s): 859152b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ subprocess.run(
15
  # Cargar el token desde las variables de entorno
16
  HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN", None)
17
 
18
- MODEL_NAME = "meta-llama/Llama-2-7b"
19
  model = AutoModelForCausalLM.from_pretrained(
20
  MODEL_NAME,
21
  torch_dtype=torch.bfloat16,
 
15
  # Cargar el token desde las variables de entorno
16
  HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN", None)
17
 
18
+ MODEL_NAME = "EleutherAI/gpt-j-6B"
19
  model = AutoModelForCausalLM.from_pretrained(
20
  MODEL_NAME,
21
  torch_dtype=torch.bfloat16,