mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-16 08:17:45 +01:00
Fix (#746)
This commit is contained in:
@@ -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',
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user