Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,10 +4,6 @@ import warnings
|
|
| 4 |
warnings.filterwarnings('ignore')
|
| 5 |
import os
|
| 6 |
|
| 7 |
-
#set the envirenment
|
| 8 |
-
os.environ["KERAS_BACKEND"] = "jax" # Or "torch" or "tensorflow".
|
| 9 |
-
|
| 10 |
-
os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
|
| 11 |
|
| 12 |
|
| 13 |
model = keras_nlp.models.CausalLM.from_preset(f"hf://soufyane/gemma_data_science")
|
|
|
|
| 4 |
warnings.filterwarnings('ignore')
|
| 5 |
import os
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
model = keras_nlp.models.CausalLM.from_preset(f"hf://soufyane/gemma_data_science")
|