Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
|
|
38 |
# bnb_4bit_compute_dtype=bfloat16
|
39 |
# )
|
40 |
|
41 |
-
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
|
42 |
model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto") # to("cuda:0")
|
43 |
terminators = [
|
44 |
tokenizer.eos_token_id,
|
|
|
38 |
# bnb_4bit_compute_dtype=bfloat16
|
39 |
# )
|
40 |
|
41 |
+
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct",token=HF_TOKEN)
|
42 |
model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto") # to("cuda:0")
|
43 |
terminators = [
|
44 |
tokenizer.eos_token_id,
|