Hugging Face's logo

Spaces:
Company270
/
LLM
Paused

App Files Community
moamen270 commited on
Commit
3cefbcb
·
1 Parent(s): c21ad85

Update endpoints.py

Browse files
Files changed (1) hide show
  1. endpoints.py +1 -1
endpoints.py CHANGED
@@ -28,7 +28,7 @@ def LLM(llm_name, length):
28
  return pipe
29
 
30
 
31
- pipe = LLM("Salesforce/codegen-16B-nl",4000)
32
  # tokenizer = AutoTokenizer.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
33
  # base_model = AutoModelForCausalLM.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
34
  # Mistral 7B
 
28
  return pipe
29
 
30
 
31
+ pipe = LLM("microsoft/phi-2",2000)
32
  # tokenizer = AutoTokenizer.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
33
  # base_model = AutoModelForCausalLM.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
34
  # Mistral 7B