Skip to content

Commit 0fc1ad3

Browse files
committed
Small tweaks
1 parent 06b599e commit 0fc1ad3

File tree

2 files changed

+7
-11
lines changed

2 files changed

+7
-11
lines changed

install.bat

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@ if "!cuda_version!" == "11.8" set useCUDA=true
2222

2323
if "!useCUDA!" == "true" (
2424

25+
call "!utilsScript!" WriteLine "Confirming VC++ Redist package installation"
26+
2527
set "RequiredVCRedistVersion=14.40.33810.00"
2628
call "!utilsScript!" GetVCredistVersion "64-bit"
2729
if "!VCredistVersion!" NEQ "!RequiredVCRedistVersion!" (

multimode_llm.py

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -163,24 +163,18 @@ def do_chat(self, user_prompt: str, image: Image, system_prompt: str=None,
163163

164164
elif use_MLX:
165165

166+
start_inference_time = time.perf_counter()
167+
166168
# Using phi_3_vision_mlx v0.0.2
167169
# https://github.com/JosefAlbers/Phi-3-Vision-MLX/tree/v0.0.2-beta
168-
start_inference_time = time.perf_counter()
169170
response = generate(self.model, self.processor, prompt, [image])
170-
inferenceMs = int((time.perf_counter() - start_inference_time) * 1000)
171171

172172
# Using latest phi_3_vision_mlx
173-
# import os
174-
# temp_name="onnx_genai_temp_image.png"
175-
# image.save(temp_name)
176-
#
177173
# agent = Agent()
178-
# start_inference_time = time.perf_counter()
179-
# response = agent(prompt, images=[temp_name])
180-
# inferenceMs = int((time.perf_counter() - start_inference_time) * 1000)
174+
# response = agent(prompt, images=[image])
181175
# agent.end()
182-
#
183-
# os.remove(temp_name)
176+
177+
inferenceMs = int((time.perf_counter() - start_inference_time) * 1000)
184178

185179
else:
186180
inputs = self.processor(prompt, image, return_tensors="pt").to(self.device)

0 commit comments

Comments
 (0)