aadya1762 commited on
Commit
6ea3b45
·
1 Parent(s): bdca525
Files changed (1) hide show
  1. gemmademo/__init__.py +2 -2
gemmademo/__init__.py CHANGED
@@ -1,11 +1,11 @@
1
  from ._chat import StreamlitChat
2
- from ._model import HuggingFaceGemmaModel
3
  from ._prompts import PromptManager
4
  from ._utils import huggingface_login
5
 
6
  __all__ = [
7
  "StreamlitChat",
8
- "HuggingFaceGemmaModel",
9
  "PromptManager",
10
  "huggingface_login",
11
  ]
 
1
  from ._chat import StreamlitChat
2
+ from ._model import LlamaCppGemmaModel
3
  from ._prompts import PromptManager
4
  from ._utils import huggingface_login
5
 
6
  __all__ = [
7
  "StreamlitChat",
8
+ "LlamaCppGemmaModel",
9
  "PromptManager",
10
  "huggingface_login",
11
  ]