Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -39,13 +39,15 @@ model_path = "ov"
|
|
39 |
pipe = ov_genai.LLMPipeline(model_path, "CPU")
|
40 |
'''
|
41 |
|
42 |
-
|
43 |
model_id,
|
44 |
-
|
|
|
45 |
)
|
46 |
|
|
|
47 |
# save the model after optimization
|
48 |
-
|
49 |
|
50 |
#pipe.start_chat()
|
51 |
|
|
|
39 |
pipe = ov_genai.LLMPipeline(model_path, "CPU")
|
40 |
'''
|
41 |
|
42 |
+
pipe = OVModelForCausalLM.from_pretrained(
|
43 |
model_id,
|
44 |
+
export=True,
|
45 |
+
quantization_config=OVWeightQuantizationConfig(bits=4),
|
46 |
)
|
47 |
|
48 |
+
|
49 |
# save the model after optimization
|
50 |
+
#pipe.save_pretrained(model_path)
|
51 |
|
52 |
#pipe.start_chat()
|
53 |
|