Spaces:
Paused
Paused
Tao Wu commited on
Commit ·
dc6ab83
1
Parent(s): 871dabe
fix bugs
Browse files- app/embedding_setup.py +2 -2
app/embedding_setup.py
CHANGED
|
@@ -107,7 +107,7 @@ def evaluate(
|
|
| 107 |
**kwargs,
|
| 108 |
)
|
| 109 |
with torch.no_grad():
|
| 110 |
-
generation_output =
|
| 111 |
**inputs,
|
| 112 |
generation_config=generation_config,
|
| 113 |
return_dict_in_generate=True,
|
|
@@ -185,7 +185,7 @@ def generate_exp(
|
|
| 185 |
**kwargs,
|
| 186 |
)
|
| 187 |
with torch.no_grad():
|
| 188 |
-
generation_output =
|
| 189 |
**inputs,
|
| 190 |
generation_config=generation_config,
|
| 191 |
return_dict_in_generate=True,
|
|
|
|
| 107 |
**kwargs,
|
| 108 |
)
|
| 109 |
with torch.no_grad():
|
| 110 |
+
generation_output = rec_adapter.generate(
|
| 111 |
**inputs,
|
| 112 |
generation_config=generation_config,
|
| 113 |
return_dict_in_generate=True,
|
|
|
|
| 185 |
**kwargs,
|
| 186 |
)
|
| 187 |
with torch.no_grad():
|
| 188 |
+
generation_output = exp_adapter.generate(
|
| 189 |
**inputs,
|
| 190 |
generation_config=generation_config,
|
| 191 |
return_dict_in_generate=True,
|