Skip to content

Commit 8dbb588

Browse files
Update packages/tasks/src/model-libraries-snippets.ts
Co-authored-by: Pedro Cuenca <[email protected]>
1 parent 97ffbe7 commit 8dbb588

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,7 @@ os.environ["KERAS_BACKEND"] = "jax"
394394
import keras_hub
395395
396396
# Load a task-specific model (*replace CausalLM with your task*)
397-
model = keras_hub.models.CausalLM.from_preset("hf://${model.id}", dtype="bloaf16)
397+
model = keras_hub.models.CausalLM.from_preset("hf://${model.id}", dtype="bfloat16")
398398
399399
# Possible tasks are CausalLM, TextToImage, ImageClassifier, ...
400400
# full list here: https://keras.io/api/keras_hub/models/#api-documentation

0 commit comments

Comments
 (0)