Hugging Face's logo

Spaces:
catallama
/
CataLlama-Chat
Sleeping

App Files Community
2
laurentiubp commited on
Commit
e83d775
·
verified ·
1 Parent(s): b721897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ if not torch.cuda.is_available():
35
 
36
 
37
  if torch.cuda.is_available():
38
- model_id = "catallama/CataLlama-v0.1-Instruct-DPO"
39
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
40
  tokenizer = AutoTokenizer.from_pretrained(model_id)
41
 
 
35
 
36
 
37
  if torch.cuda.is_available():
38
+ model_id = "catallama/CataLlama-v0.2-Instruct-SFT"
39
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
40
  tokenizer = AutoTokenizer.from_pretrained(model_id)
41