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