Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import glob
|
|
| 2 |
import gradio as gr
|
| 3 |
import tempfile
|
| 4 |
import os
|
| 5 |
-
import
|
| 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 =
|
| 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,
|