Fix the RNG is not passed into pipelineConfig.

1 year ago · a07e69ac5d
parent 09602864d6
commit a07e69ac5d
1 changed files with 1 additions and 0 deletions
--- a/swift/StableDiffusionCLI/main.swift
+++ b/swift/StableDiffusionCLI/main.swift
@ -129,6 +129,7 @@ struct StableDiffusionSample: ParsableCommand {
        pipelineConfig.seed = seed
        pipelineConfig.guidanceScale = guidanceScale
        pipelineConfig.schedulerType = scheduler.stableDiffusionScheduler
+        pipelineConfig.rngType = rng.stableDiffusionRNG
        
        let images = try pipeline.generateImages(
            configuration: pipelineConfig,