Hugging Face's logo

Spaces:
yuchenlin
/
MagpieLM-8B
Running on Zero

App Files Community
flydust commited on
Commit
8b1642c
•
1 Parent(s): 5450145

update requirements

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import spaces
6
  model_name = "Magpie-Align/Llama-3-8B-Magpie-Align-v0.1"
7
 
8
  device = "cuda" # the device to load the model onto
9
- tokenizer = AutoTokenizer.from_pretrained(model_name)
10
  model = AutoModelForCausalLM.from_pretrained(
11
  model_name,
12
  torch_dtype="auto"
 
6
  model_name = "Magpie-Align/Llama-3-8B-Magpie-Align-v0.1"
7
 
8
  device = "cuda" # the device to load the model onto
9
+ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
10
  model = AutoModelForCausalLM.from_pretrained(
11
  model_name,
12
  torch_dtype="auto"
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- huggingface_hub==0.23.2
2
  transformers==4.42.3
3
  spaces==0.28.3
4
- numpy==1.26.4
 
1
+ huggingface_hub==0.22.2
2
  transformers==4.42.3
3
  spaces==0.28.3
4
+ bm25s==0.1.7