badaoui HF Staff commited on
Commit
f0be84b
·
verified ·
1 Parent(s): cc6b586

Update optimum_neuron_export.py

Browse files
Files changed (1) hide show
  1. 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")