Spaces:
Paused
Paused
Update endpoints.py
Browse files- endpoints.py +1 -0
endpoints.py
CHANGED
|
@@ -28,6 +28,7 @@ app.add_middleware(
|
|
| 28 |
|
| 29 |
|
| 30 |
def LLM(llm_name, length):
|
|
|
|
| 31 |
tokenizer = AutoTokenizer.from_pretrained(llm_name)
|
| 32 |
model = AutoModelForCausalLM.from_pretrained(llm_name)
|
| 33 |
pipe = pipeline("text-generation",
|
|
|
|
| 28 |
|
| 29 |
|
| 30 |
def LLM(llm_name, length):
|
| 31 |
+
print(llm_name)
|
| 32 |
tokenizer = AutoTokenizer.from_pretrained(llm_name)
|
| 33 |
model = AutoModelForCausalLM.from_pretrained(llm_name)
|
| 34 |
pipe = pipeline("text-generation",
|