Spaces:
Sleeping
Sleeping
Update connections/model.py
Browse files- connections/model.py +3 -2
connections/model.py
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
-
from unsloth import FastLanguageModel
|
2 |
import spaces
|
3 |
-
|
4 |
@spaces.GPU
|
|
|
|
|
|
|
5 |
class InferencePipeline:
|
6 |
def __init__(self, conf, api_key):
|
7 |
self.conf = conf
|
|
|
|
|
1 |
import spaces
|
|
|
2 |
@spaces.GPU
|
3 |
+
from unsloth import FastLanguageModel
|
4 |
+
|
5 |
+
|
6 |
class InferencePipeline:
|
7 |
def __init__(self, conf, api_key):
|
8 |
self.conf = conf
|