moamen270 commited on
Commit
855b075
·
1 Parent(s): 85903cb

Update endpoints.py

Browse files
Files changed (1) hide show
  1. endpoints.py +1 -1
endpoints.py CHANGED
@@ -30,7 +30,7 @@ import requests
30
  # Use a pipeline as a high-level helper
31
  from transformers import pipeline
32
 
33
- pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
34
 
35
  # tokenizer = AutoTokenizer.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
36
  # base_model = AutoModelForCausalLM.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
 
30
  # Use a pipeline as a high-level helper
31
  from transformers import pipeline
32
 
33
+ pipe = pipeline("text-generation", model="codellama/CodeLlama-7b-hf")
34
 
35
  # tokenizer = AutoTokenizer.from_pretrained("WizardLM/WizardCoder-1B-V1.0")
36
  # base_model = AutoModelForCausalLM.from_pretrained("WizardLM/WizardCoder-1B-V1.0")