fix progress #79 (#83)

huggingface · Mar 28, 2024 · 74b9421 · 74b9421
1 parent 9d82e00
commit 74b9421
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/Hub/Downloader.swift b/Sources/Hub/Downloader.swift
@@ -105,8 +105,8 @@ class Downloader: NSObject, ObservableObject {
 }
 
 extension Downloader: URLSessionDownloadDelegate {
-    func urlSession(_: URLSession, downloadTask: URLSessionDownloadTask, didWriteData _: Int64, totalBytesWritten _: Int64, totalBytesExpectedToWrite _: Int64) {
-        downloadState.value = .downloading(downloadTask.progress.fractionCompleted)
+    func urlSession(_: URLSession, downloadTask: URLSessionDownloadTask, didWriteData _: Int64, totalBytesWritten: Int64, totalBytesExpectedToWrite: Int64) {
+        downloadState.value = .downloading(Double(totalBytesWritten) / Double(totalBytesExpectedToWrite))
     }
 
     func urlSession(_: URLSession, downloadTask _: URLSessionDownloadTask, didFinishDownloadingTo location: URL) {