Spaces:
Sleeping
Sleeping
Update services/llama_generator.py
Browse files
services/llama_generator.py
CHANGED
@@ -4,7 +4,7 @@ from config.config import GenerationConfig, ModelConfig
|
|
4 |
from typing import List, Dict, Any, Optional, Tuple
|
5 |
from datetime import datetime
|
6 |
import logging
|
7 |
-
import
|
8 |
|
9 |
from config.config import settings
|
10 |
|
|
|
4 |
from typing import List, Dict, Any, Optional, Tuple
|
5 |
from datetime import datetime
|
6 |
import logging
|
7 |
+
import torch
|
8 |
|
9 |
from config.config import settings
|
10 |
|