Moibe commited on
Commit
452fa91
Β·
1 Parent(s): 5c122eb

Inference verbose

Browse files
Files changed (1) hide show
  1. funciones.py +4 -2
funciones.py CHANGED
@@ -8,7 +8,7 @@ def genera_platillo_gpu(platillo):
8
 
9
  prompt = globales.previo + platillo
10
 
11
- print("Eso es el prompt final:", prompt)
12
 
13
  kwargs = {
14
  "prompt": prompt,
@@ -33,7 +33,7 @@ def genera_platillo_gpu(platillo):
33
  #Cuando es GPU, debe de restar segundos disponibles de HF
34
  herramientas.restaSegundosGPU(globales.work_cost)
35
 
36
- print("Lo que estoy regresando es result[0] o sea: ", result[0])
37
 
38
  return result[0]
39
 
@@ -46,6 +46,8 @@ def genera_platillo_gpu(platillo):
46
 
47
  def genera_platillo_inference(platillo):
48
 
 
 
49
  client = InferenceClient(
50
  provider= globales.proveedor,
51
  api_key=globales.llave
 
8
 
9
  prompt = globales.previo + platillo
10
 
11
+ print("Platillo generado:", platillo)
12
 
13
  kwargs = {
14
  "prompt": prompt,
 
33
  #Cuando es GPU, debe de restar segundos disponibles de HF
34
  herramientas.restaSegundosGPU(globales.work_cost)
35
 
36
+ print("Resultado regresado en result[0] es: ", result[0])
37
 
38
  return result[0]
39
 
 
46
 
47
  def genera_platillo_inference(platillo):
48
 
49
+ print("Platillo generado:", platillo)
50
+
51
  client = InferenceClient(
52
  provider= globales.proveedor,
53
  api_key=globales.llave