cryptocalypse commited on
Commit
fd22519
verified
1 Parent(s): 8e7b8ef

Update gen.py

Browse files
Files changed (1) hide show
  1. gen.py +1 -1
gen.py CHANGED
@@ -4,7 +4,7 @@ import sys
4
  import sys
5
 
6
  # Cargar el pipeline de generaci贸n de texto
7
- pipe = pipeline("text-generation", model="google/gemma-7b", torch_dtype=torch.bfloat16, device_map="auto")
8
 
9
  # Definir el prompt para generar un JSON con eventos anidados
10
  prompt = (
 
4
  import sys
5
 
6
  # Cargar el pipeline de generaci贸n de texto
7
+ pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-alpha", torch_dtype=torch.bfloat16, device_map="auto")
8
 
9
  # Definir el prompt para generar un JSON con eventos anidados
10
  prompt = (