Spaces:

bmas10
/

TestLlama2

Sleeping

bmas10 commited on Feb 19

Commit

edf589b

verified ·

1 Parent(s): 7e7a085

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,10 +11,11 @@ import torch
 from transformers import pipeline
 model_name = "bmas10/DeepSeek-Llama-8-GGUF"
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
-pipe = pipeline("text-generation", model=model_name)
 pipe(messages)

 from transformers import pipeline
 model_name = "bmas10/DeepSeek-Llama-8-GGUF"
+model = AutoModel.from_pretrained(model_name)
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
+pipe = pipeline("text-generation", model=model)
 pipe(messages)