1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
34repo ="gustavecortal/Beck-0.6B"56tokenizer = AutoTokenizer.from_pretrained(model_path)7model = AutoModelForCausalLM.from_pretrained(8 model_path,9 torch_dtype=torch.bfloat16,10 device_map="cuda",11 low_cpu_mem_usage=True,12)13model.eval()1415prompt =r"""Anxiety often strikes when I'm faced with making decisions.
16 The fear of making the wrong choice or disappointing others paralyzes me, leaving me stuck in indecision.
17 I want to learn how to trust myself and make confident choices. \no_think"""18messages_think =[19{"role":"system","content":"Do not use bullets points."},20{"role":"user","content": prompt}21]2223text = tokenizer.apply_chat_template(24 messages_think,25 tokenize=False,26 add_generation_prompt=True,27)28model_inputs = tokenizer([text], return_tensors="pt").to(model.device)2930with torch.no_grad():31 generated_ids = model.generate(32**model_inputs,33 max_new_tokens=2048,34 do_sample=True,35 temperature=1.0,36 top_p=0.95,37 top_k=50,38 repetition_penalty=1.1,39 eos_token_id=tokenizer.eos_token_id,40 pad_token_id=tokenizer.eos_token_id,41 use_cache=True,42)4344output_ids = generated_ids[0][len(model_inputs.input_ids[0]):]45print(tokenizer.decode(output_ids, skip_special_tokens=True))
How does it compare to other language models?
For the given prompt:
Anxiety often strikes when I'm faced with making decisions.
The fear of making the wrong choice or disappointing others paralyzes me, leaving me stuck in indecision.
I want to learn how to trust myself and make confident choices. \no_think
Qwen3-0.6B:
It's important to recognize that anxiety about decision-making is common and often rooted in past experiences or fears. When you feel stuck, it might be helpful to take a moment to breathe deeply or reflect on what kind of person you are. Trusting yourself involves practicing self-compassion—being kind to yourself even when you don't have confidence. Making confident decisions requires taking small steps towards building that confidence each day.
If these thoughts resonate with you, consider talking to someone who understands your feelings, whether they are close friends, family, or a professional counselor. Small steps can gradually build confidence, so even if you're unsure yet, consistent effort is key. Let this remind you that progress is possible every day.
Beck-0.6B:
It takes courage to acknowledge when you're feeling overwhelmed by anxiety about making decisions. It sounds like you're going through a time when confidence is really being tested.
Firstly, it's great that you recognize these feelings and seek help. This is a step towards healing and growth. Can you tell me more about what specific situations cause you stress when considering decisions? Are there any particular thoughts or fears that come up for you as you approach each decision?
We can work together to identify patterns in your anxiety so we can understand why you're struggling to make confident choices. By addressing those underlying anxieties, we might be able to develop coping strategies that help you feel more grounded and empowered. Remember, mistakes are valuable learning experiences. When you choose wisely, they become opportunities to grow and learn from your decisions. Would that resonate with you?
Methodology
Beck was trained using preference optimization (ORPO) and LoRA. You can reproduce the results using my repo for lightweight preference optimization using this config that contains the hyperparameters.
This work was performed using HPC resources (Jean Zay supercomputer) from GENCI-IDRIS (Grant 20XX-AD011014205).
Inspiration
Beck aims to reason about psychological and philosophical concepts such as self-image, emotion, and existence.
