Spaces:
Paused
Paused
Update endpoints.py
Browse files- 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="
|
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")
|