Spaces:
Running
Running
chore: adding HF_TOKEN
Browse files
app.py
CHANGED
@@ -5,6 +5,8 @@ import torch
|
|
5 |
model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
|
6 |
device_map = 'cuda'
|
7 |
|
|
|
|
|
8 |
def load_model() -> AutoModelForCausalLM:
|
9 |
return AutoModelForCausalLM.from_pretrained(model_name, device_map=device_map)
|
10 |
|
|
|
5 |
model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
|
6 |
device_map = 'cuda'
|
7 |
|
8 |
+
HF_TOKEN = os.environ.get("HF_TOKEN", None)
|
9 |
+
|
10 |
def load_model() -> AutoModelForCausalLM:
|
11 |
return AutoModelForCausalLM.from_pretrained(model_name, device_map=device_map)
|
12 |
|