Downloads · 30 days
3
5% of all-time downloads
gemma_mental_health_2b_en
tatihden/gemma_mental_health_2b_en
gemma_mental_health_2b_en is a text generation model from tatihden. Use it when you need the model to write or continue text. It is set up for keras-hub.
This is a Gemma model uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends. This model is related to a CausalLM task.
All-time downloads
59
Public
Repo size
10 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.h510 GB · 100%
At a glance
- Task
- Text Generation
- Library
- keras-hub
- Access
- Public
- Created
- Nov 14, 2024
- Updated
- Nov 14, 2024
- SHA
- 84faa789
Try a prompt
Spaces
Model card
From the Hugging Face model README
This is a Gemma model uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
This model is related to a CausalLM task.
Model config:
- name: gemma_backbone
- trainable: True
- vocabulary_size: 256000
- num_layers: 18
- num_query_heads: 8
- num_key_value_heads: 1
- hidden_dim: 2048
- intermediate_dim: 32768
- head_dim: 256
- layer_norm_epsilon: 1e-06
- dropout: 0
- query_head_dim_normalize: True
- use_post_ffw_norm: False
- use_post_attention_norm: False
- final_logit_soft_cap: None
- attention_logit_soft_cap: None
- sliding_window_size: 4096
- use_sliding_window_attention: False
This model card has been generated automatically and should be completed by the model author. See Model Cards documentation for more information.
- Task
- Text Generation
- Library
- keras-hub
- Created
- Nov 14, 2024
- Updated
- Nov 14, 2024