Hugging Face's logo

Spaces:
K00B404
/
TeachingPadawan
Runtime error

App Files Community
K00B404 commited on
Commit
eb33590
•
1 Parent(s): ce72ab2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
  # Load your teacher model
6
- model_name = "0x0mom/nous_gemma_r1"# "cognitivecomputations/dolphin-2_6-phi-2" # "Dizzykong/gpt2-medium-commands"#
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9
 
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
  # Load your teacher model
6
+ model_name = "microsoft/Orca-2-7b" #"0x0mom/nous_gemma_r1"# "cognitivecomputations/dolphin-2_6-phi-2" # "Dizzykong/gpt2-medium-commands"#
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9