Reverted identation change

pull/46/head
Wanaldino Antimonio 3 years ago
parent c535f219ff
commit 4caa5e3027

@ -55,13 +55,11 @@ public struct StableDiffusionPipeline: ResourceManaging {
/// - safetyChecker: Optional model for checking safety of generated images /// - safetyChecker: Optional model for checking safety of generated images
/// - reduceMemory: Option to enable reduced memory mode /// - reduceMemory: Option to enable reduced memory mode
/// - Returns: Pipeline ready for image generation /// - Returns: Pipeline ready for image generation
public init( public init(textEncoder: TextEncoder,
textEncoder: TextEncoder,
unet: Unet, unet: Unet,
decoder: Decoder, decoder: Decoder,
safetyChecker: SafetyChecker? = nil, safetyChecker: SafetyChecker? = nil,
reduceMemory: Bool = false reduceMemory: Bool = false) {
) {
self.textEncoder = textEncoder self.textEncoder = textEncoder
self.unet = unet self.unet = unet
self.decoder = decoder self.decoder = decoder

Loading…
Cancel
Save