Spaces:
Sleeping
Sleeping
cryptocalypse
commited on
Update gen.py
Browse files
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="
|
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 = (
|