Hugging Face's logo

Spaces:
zidsi
/
SLlamica_test
Sleeping

App Files Community
zidsi commited on
Commit
b244f76
·
1 Parent(s): 4983f9e

back to v2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
7
  import os
8
  HF_TOKEN = os.getenv('HF_TOKEN')
9
 
10
- checkpoint = "zidsi/SLlamica_PT4SFT_v1"
11
  device = "cuda" # "cuda" or "cpu"
12
  tokenizer = AutoTokenizer.from_pretrained(checkpoint,token=HF_TOKEN)
13
  model = AutoModelForCausalLM.from_pretrained(checkpoint,token=HF_TOKEN)
 
7
  import os
8
  HF_TOKEN = os.getenv('HF_TOKEN')
9
 
10
+ checkpoint = "zidsi/SLlamica_PT4SFT_v2"
11
  device = "cuda" # "cuda" or "cpu"
12
  tokenizer = AutoTokenizer.from_pretrained(checkpoint,token=HF_TOKEN)
13
  model = AutoModelForCausalLM.from_pretrained(checkpoint,token=HF_TOKEN)