Migrations
/
ml-stable-diffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Accept suggestion from code review

Browse Source 
Co-authored-by: Alejandro Isaza <167236+alejandro-isaza@users.noreply.github.com>
pull/59/head
Pedro Cuenca 3 years ago committed by GitHub
parent 7b4c891147
commit f621a89ccc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
swift/StableDiffusion/pipeline/Scheduler.swift
Unescape View File

@ -55,7 +55,7 @@ public extension Scheduler {
/// - Returns: sum_i weights[i]*values[i] /// - Returns: sum_i weights[i]*values[i]
func weightedSum(_ weights: [Double], _ values: [MLShapedArray<Float32>]) -> MLShapedArray<Float32> { func weightedSum(_ weights: [Double], _ values: [MLShapedArray<Float32>]) -> MLShapedArray<Float32> {
assert(weights.count > 1 && values.count == weights.count) assert(weights.count > 1 && values.count == weights.count)
assert(values.allSatisfy({$0.scalarCount == values.first!.scalarCount})) assert(values.allSatisfy({ $0.scalarCount == values.first!.scalarCount }))
var w = Float(weights.first!) var w = Float(weights.first!)
var scalars = values.first!.scalars.map({ $0 * w }) var scalars = values.first!.scalars.map({ $0 * w })
for next in 1 ..< values.count { for next in 1 ..< values.count {

Write Preview
Loading…
Cancel
Save