tykiww commited on
Commit
e4d8995
1 Parent(s): 78210ab

Update connections/model.py

Browse files
Files changed (1) hide show
  1. 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