Spaces:
Runtime error
Runtime error
storresbusquets
commited on
Commit
•
26bf4a5
1
Parent(s):
999acec
Update app.py
Browse files
app.py
CHANGED
@@ -9,17 +9,17 @@ access_token = os.getenv("Llama2")
|
|
9 |
|
10 |
def greet(text):
|
11 |
|
12 |
-
model = "
|
13 |
tokenizer = AutoTokenizer.from_pretrained(model, token=access_token)
|
14 |
|
15 |
pipeline = transformers.pipeline(
|
16 |
"text-generation",
|
17 |
model=model,
|
18 |
tokenizer=tokenizer,
|
19 |
-
torch_dtype=torch.bfloat16,
|
20 |
trust_remote_code=True,
|
21 |
device_map="auto",
|
22 |
-
max_length=
|
23 |
do_sample=True,
|
24 |
top_k=10,
|
25 |
num_return_sequences=1,
|
|
|
9 |
|
10 |
def greet(text):
|
11 |
|
12 |
+
model = "meta-llama/Llama-2-7b-hf"
|
13 |
tokenizer = AutoTokenizer.from_pretrained(model, token=access_token)
|
14 |
|
15 |
pipeline = transformers.pipeline(
|
16 |
"text-generation",
|
17 |
model=model,
|
18 |
tokenizer=tokenizer,
|
19 |
+
# torch_dtype=torch.bfloat16,
|
20 |
trust_remote_code=True,
|
21 |
device_map="auto",
|
22 |
+
max_length=512,
|
23 |
do_sample=True,
|
24 |
top_k=10,
|
25 |
num_return_sequences=1,
|