dgarrett-synaptics commited on
Commit
2015cbd
·
verified ·
1 Parent(s): d8e3c5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import glob
2
  import gradio as gr
3
  import tempfile
4
  import os
5
- import sr100_model_compiler
6
  import html
7
  import pathlib
8
  import spaces
@@ -71,7 +71,7 @@ def compile_model(model_name, vmem_value, lpmem_value, uploaded_model):
71
  print(f"Starting model optimization for {temp_model_path}")
72
  print(f"VMEM limit: {vmem_size_limit}, LPMEM limit: {lpmem_size_limit}")
73
 
74
- success, results = sr100_model_compiler.sr100_model_optimizer(
75
  model_file=temp_model_path,
76
  vmem_size_limit=vmem_size_limit,
77
  lpmem_size_limit=lpmem_size_limit,
 
2
  import gradio as gr
3
  import tempfile
4
  import os
5
+ import sr_model_compiler
6
  import html
7
  import pathlib
8
  import spaces
 
71
  print(f"Starting model optimization for {temp_model_path}")
72
  print(f"VMEM limit: {vmem_size_limit}, LPMEM limit: {lpmem_size_limit}")
73
 
74
+ success, results = sr_model_compiler.sr100_model_optimizer(
75
  model_file=temp_model_path,
76
  vmem_size_limit=vmem_size_limit,
77
  lpmem_size_limit=lpmem_size_limit,