Spaces:
Running
Running
Update optimum_neuron_export.py
Browse files- optimum_neuron_export.py +1 -1
optimum_neuron_export.py
CHANGED
|
@@ -55,7 +55,7 @@ from optimum.neuron import (
|
|
| 55 |
|
| 56 |
from optimum.neuron.cache import synchronize_hub_cache
|
| 57 |
from synchronizer import synchronize_hub_cache_with_pr
|
| 58 |
-
from optimum.exporters.neuron import main_export, build_stable_diffusion_components_mandatory_shapes
|
| 59 |
|
| 60 |
SPACES_URL = "https://huggingface.co/spaces/optimum/neuron-export"
|
| 61 |
CUSTOM_CACHE_REPO = os.getenv("CUSTOM_CACHE_REPO")
|
|
|
|
| 55 |
|
| 56 |
from optimum.neuron.cache import synchronize_hub_cache
|
| 57 |
from synchronizer import synchronize_hub_cache_with_pr
|
| 58 |
+
from optimum.exporters.neuron import main_export, build_stable_diffusion_components_mandatory_shapes
|
| 59 |
|
| 60 |
SPACES_URL = "https://huggingface.co/spaces/optimum/neuron-export"
|
| 61 |
CUSTOM_CACHE_REPO = os.getenv("CUSTOM_CACHE_REPO")
|