ntaexams commited on
Commit
3905f39
·
verified ·
1 Parent(s): e653420

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -4,6 +4,11 @@ import torch
4
 
5
  # Load the model and tokenizer
6
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 
 
 
7
  import torch
8
  import bitsandbytes as bnb
9
 
 
4
 
5
  # Load the model and tokenizer
6
  from transformers import AutoModelForCausalLM, AutoTokenizer
7
+ from llama_cpp import Llama
8
+
9
+ model = Llama(model_path="phi-2.Q4_K_M.gguf") # Use a quantized model
10
+ output = model("Solve the equation x^2 - 5x + 6 = 0")
11
+ print(output)
12
  import torch
13
  import bitsandbytes as bnb
14