Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def load_quantized_model(model_id, model_basename):
|
|
35 |
|
36 |
|
37 |
# Making the code device-agnostic
|
38 |
-
|
39 |
|
40 |
|
41 |
|
@@ -55,7 +55,7 @@ def load_model_norm():
|
|
55 |
return model, tokenizer
|
56 |
|
57 |
|
58 |
-
model, tokenizer = load_model_norm()
|
59 |
|
60 |
|
61 |
# Function to generate a response using the model
|
|
|
35 |
|
36 |
|
37 |
# Making the code device-agnostic
|
38 |
+
model, tokenizer = load_quantized_model(model_name_or_path, "model.safetensors")
|
39 |
|
40 |
|
41 |
|
|
|
55 |
return model, tokenizer
|
56 |
|
57 |
|
58 |
+
#model, tokenizer = load_model_norm()
|
59 |
|
60 |
|
61 |
# Function to generate a response using the model
|