Spaces:
Sleeping
Sleeping
trying keras
Browse files- backend.py +2 -2
- interface.py +1 -1
backend.py
CHANGED
@@ -25,8 +25,8 @@ os.environ["KERAS_BACKEND"] = "jax"
|
|
25 |
os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "1.00"
|
26 |
|
27 |
|
28 |
-
os.getenv("KAGGLE_USERNAME")
|
29 |
-
os.getenv["KAGGLE_KEY"]
|
30 |
|
31 |
"""huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
32 |
login(huggingface_token)
|
|
|
25 |
os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "1.00"
|
26 |
|
27 |
|
28 |
+
#os.getenv("KAGGLE_USERNAME")
|
29 |
+
#os.getenv["KAGGLE_KEY"]
|
30 |
|
31 |
"""huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
32 |
login(huggingface_token)
|
interface.py
CHANGED
@@ -69,7 +69,7 @@ import keras_nlp
|
|
69 |
if not streamed_response:
|
70 |
yield CompletionResponse(text="No response generated.", delta="No response generated.")"""
|
71 |
|
72 |
-
|
73 |
class GemmaLLMInterface(CustomLLM):
|
74 |
model: keras_nlp.models.GemmaCausalLM = None
|
75 |
context_window: int = 8192
|
|
|
69 |
if not streamed_response:
|
70 |
yield CompletionResponse(text="No response generated.", delta="No response generated.")"""
|
71 |
|
72 |
+
# for Keras
|
73 |
class GemmaLLMInterface(CustomLLM):
|
74 |
model: keras_nlp.models.GemmaCausalLM = None
|
75 |
context_window: int = 8192
|