Spaces:
Paused
Paused
Update app_wip.py
Browse files- app_wip.py +1 -1
app_wip.py
CHANGED
|
@@ -15,7 +15,7 @@ from pipeline import (
|
|
| 15 |
CausalInferencePipeline,
|
| 16 |
)
|
| 17 |
from utils.dataset import TextDataset
|
| 18 |
-
|
| 19 |
from demo_utils.memory import get_cuda_free_memory_gb, DynamicSwapInstaller
|
| 20 |
|
| 21 |
# -------------------------------------------------------------------
|
|
|
|
| 15 |
CausalInferencePipeline,
|
| 16 |
)
|
| 17 |
from utils.dataset import TextDataset
|
| 18 |
+
from utils.misc import set_seed
|
| 19 |
from demo_utils.memory import get_cuda_free_memory_gb, DynamicSwapInstaller
|
| 20 |
|
| 21 |
# -------------------------------------------------------------------
|