llm_lib 0.1.2 → 0.2.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/lib/llm_lib.rb +12 -4
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 40b5f225218a674034c99e849437850386407b735cc20b78bd0139fc5be37d91
|
4
|
+
data.tar.gz: 41032e97b353a8a1c1cfe8cee0211be41035688fb2475adbe113fa5064eb7ee1
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 418f302f026fefd27f857a87172fc044847206cb0bbce838315595aa93ec86e4b62edf00d6cdfdd9ab85eb65580284bcc3658f013da4b4f53b9fa081d4d91188
|
7
|
+
data.tar.gz: 55840841a1978aed41e7165ea40c7ac13fdd1da79f38f5b6c55151a1571eba90bfed53b818c2bbe9f5f5b385aee20615e7603f4bf8ce0e919b2e6ef6d5106640
|
data/lib/llm_lib.rb
CHANGED
@@ -75,15 +75,23 @@ module LlmLib
|
|
75
75
|
|
76
76
|
def hugging_falcon_call(query, model = "tiiuae/falcon-40b-instruct")
|
77
77
|
response = HuggingFace.send(@api_key,
|
78
|
-
|
79
|
-
|
78
|
+
model,
|
79
|
+
query)
|
80
80
|
response
|
81
81
|
end
|
82
82
|
|
83
83
|
def hugging_llama2_call(query, model = "meta-llama/Llama-2-70b-chat-hf")
|
84
84
|
response = HuggingFace.send(@api_key,
|
85
|
-
|
86
|
-
|
85
|
+
model,
|
86
|
+
query)
|
87
|
+
response
|
88
|
+
end
|
89
|
+
|
90
|
+
def hugging_dolly2_call(query, model = "databricks/dolly-v2-12b")
|
91
|
+
response = HuggingFace.send(@api_key,
|
92
|
+
model,
|
93
|
+
query
|
94
|
+
)
|
87
95
|
response
|
88
96
|
end
|
89
97
|
|