Spaces:
Runtime error
Runtime error
gorkemgoknar
commited on
Commit
•
3628ba1
1
Parent(s):
3e956eb
Update app.py
Browse files
app.py
CHANGED
@@ -26,6 +26,7 @@ STOP_LIST=["###","##"]
|
|
26 |
|
27 |
stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=64)])
|
28 |
|
|
|
29 |
llm = AutoModelForCausalLM.from_pretrained("gorkemgoknar/llama2-7f-moviechatbot-ggml-q4",
|
30 |
model_type='llama',
|
31 |
gpu_layers=GPU_LAYERS,
|
|
|
26 |
|
27 |
stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=64)])
|
28 |
|
29 |
+
|
30 |
llm = AutoModelForCausalLM.from_pretrained("gorkemgoknar/llama2-7f-moviechatbot-ggml-q4",
|
31 |
model_type='llama',
|
32 |
gpu_layers=GPU_LAYERS,
|