Chris4K commited on
Commit
c4f4841
·
verified ·
1 Parent(s): e4efac1

Update services/llama_generator.py

Browse files
Files changed (1) hide show
  1. services/llama_generator.py +1 -1
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 pytorch
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