sultan-hassan commited on
Commit
52919bc
·
verified ·
1 Parent(s): 74ac718

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ body::before {
35
  }
36
  """
37
 
38
-
39
  gemma_lm = keras_nlp.models.CausalLM.from_preset("hf://sultan-hassan/CosmoGemma_2b_en")
40
 
41
  def launch(input):
 
35
  }
36
  """
37
 
38
+ keras.config.set_floatx("bfloat16")
39
  gemma_lm = keras_nlp.models.CausalLM.from_preset("hf://sultan-hassan/CosmoGemma_2b_en")
40
 
41
  def launch(input):