This commit is contained in:
co63oc
2024-02-21 15:59:37 +08:00
committed by GitHub
parent 10e98f166f
commit 07a5bef0ca
3 changed files with 5 additions and 5 deletions

View File

@@ -8,7 +8,7 @@ if TYPE_CHECKING:
from .base import Exporter
from .builder import build_exporter
from .cv import CartoonTranslationExporter, FaceDetectionSCRFDExporter
from .multi_modal import StableDiffuisonExporter
from .multi_modal import StableDiffusionExporter
from .nlp import (CsanmtForTranslationExporter,
SbertForSequenceClassificationExporter,
SbertForZeroShotClassificationExporter)
@@ -19,7 +19,7 @@ else:
'base': ['Exporter'],
'builder': ['build_exporter'],
'cv': ['CartoonTranslationExporter', 'FaceDetectionSCRFDExporter'],
'multi_modal': ['StableDiffuisonExporter'],
'multi_modal': ['StableDiffusionExporter'],
'nlp': [
'CsanmtForTranslationExporter',
'SbertForSequenceClassificationExporter',

View File

@@ -5,10 +5,10 @@ from typing import TYPE_CHECKING
from modelscope.utils.import_utils import LazyImportModule
if TYPE_CHECKING:
from .stable_diffusion_export import StableDiffuisonExporter
from .stable_diffusion_export import StableDiffusionExporter
else:
_import_structure = {
'stable_diffusion_export': ['StableDiffuisonExporter'],
'stable_diffusion_export': ['StableDiffusionExporter'],
}
import sys

View File

@@ -23,7 +23,7 @@ from modelscope.utils.hub import snapshot_download
@EXPORTERS.register_module(
Tasks.text_to_image_synthesis, module_name=Models.stable_diffusion)
class StableDiffuisonExporter(TorchModelExporter):
class StableDiffusionExporter(TorchModelExporter):
@torch.no_grad()
def export_onnx(self,