Spaces:
Paused
Paused
Commit
•
a1dfc9d
1
Parent(s):
ff65288
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ h1 {
|
|
55 |
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2",use_auth_token=access_token)
|
56 |
model = PeftModel.from_pretrained(model, "physician-ai/mistral-finetuned1",use_auth_token=access_token)
|
57 |
tokenizer = AutoTokenizer.from_pretrained("physician-ai/mistral-finetuned1",use_auth_token=access_token)
|
58 |
-
text_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=
|
59 |
|
60 |
terminators = [
|
61 |
tokenizer.eos_token_id,
|
|
|
55 |
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2",use_auth_token=access_token)
|
56 |
model = PeftModel.from_pretrained(model, "physician-ai/mistral-finetuned1",use_auth_token=access_token)
|
57 |
tokenizer = AutoTokenizer.from_pretrained("physician-ai/mistral-finetuned1",use_auth_token=access_token)
|
58 |
+
text_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512, temperature=0.8, top_p=0.95, repetition_penalty=1.15)
|
59 |
|
60 |
terminators = [
|
61 |
tokenizer.eos_token_id,
|