Update inferencer.py
Browse files- inferencer.py +1 -0
inferencer.py
CHANGED
|
@@ -82,6 +82,7 @@ class Inferencer:
|
|
| 82 |
self.cfg_prompt = cfg_prompt
|
| 83 |
self.model = self.init_model()
|
| 84 |
|
|
|
|
| 85 |
def init_model(self):
|
| 86 |
# config = Config.fromfile(self.config_file)
|
| 87 |
# model = BUILDER.build(config.model)
|
|
|
|
| 82 |
self.cfg_prompt = cfg_prompt
|
| 83 |
self.model = self.init_model()
|
| 84 |
|
| 85 |
+
@spaces.GPU
|
| 86 |
def init_model(self):
|
| 87 |
# config = Config.fromfile(self.config_file)
|
| 88 |
# model = BUILDER.build(config.model)
|