Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -22,8 +22,6 @@ def generate(
|
|
22 |
):
|
23 |
try:
|
24 |
pipe = load_model(model_name)
|
25 |
-
if model_name == "Locutusque/Hercules-6.1-Llama-3.1-8B":
|
26 |
-
model_name = "meta-llama/Llama-3.1-8B-Instruct"
|
27 |
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, token=os.environ["token"])
|
28 |
tokenizer.eos_token = "<|im_end|>"
|
29 |
print(tokenizer)
|
|
|
22 |
):
|
23 |
try:
|
24 |
pipe = load_model(model_name)
|
|
|
|
|
25 |
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, token=os.environ["token"])
|
26 |
tokenizer.eos_token = "<|im_end|>"
|
27 |
print(tokenizer)
|