Merge pull request #125 from ionite34/merge-fork

.github/workflows/cla.yml

@@ -0,0 +1,38 @@
+name: "CLA Assistant"
+
+on:
+  issue_comment:
+    types: [created]
+  pull_request_target:
+    types: [opened,closed,synchronize]
+
+# explicitly configure permissions, in case your GITHUB_TOKEN workflow permissions are set to read-only in repository settings
+permissions:
+  actions: write
+  contents: write
+  pull-requests: write
+  statuses: write
+
+jobs:
+  CLAAssistant:
+    if: github.repository == 'LykosAI/StabilityMatrix'
+    runs-on: ubuntu-latest
+    steps:
+      - name: "CLA Assistant"
+        if: (github.event.comment.body == 'recheck' || github.event.comment.body == 'I have read the CLA Document and I hereby sign the CLA') || github.event_name == 'pull_request_target'
+        uses: contributor-assistant/github-action@v2.3.0
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          # the below token should have repo scope and must be manually added by you in the repository's secret
+          # This token is required only if you have configured to store the signatures in a remote repository/organization
+          PERSONAL_ACCESS_TOKEN: ${{ secrets.CLA_CONFIG_ACCESS_TOKEN }}
+        with:
+          path-to-signatures: 'signatures/version1/cla.json'
+          path-to-document: 'https://lykos.ai/cla'
+          # branch should not be protected
+          branch: 'main'
+          allowlist: ionite34,mohnjiles,bot*
+          # the followings are the optional inputs - If the optional inputs are not given, then default values will be taken
+          remote-organization-name: LykosAI
+          remote-repository-name: clabot-config
+          custom-allsigned-prcomment: '**CLA Assistant bot** All Contributors have signed the CLA.'

StabilityMatrix/CheckpointBrowserPage.xaml

@@ -113,9 +113,15 @@
                             Appearance="Primary"
                             Command="{Binding ImportCommand}"
                             CommandParameter="{Binding CivitModel}"
-                            Content="Import"
                             HorizontalAlignment="Stretch"
-                            Margin="0,8,0,0" />
+                            Margin="0,8,0,0">
+                            <ui:Button.Content>
+                                <StackPanel Orientation="Horizontal">
+                                    <TextBlock Text="Import - "/>
+                                    <TextBlock Text="{Binding CivitModel.FullFilesSize}"/>
+                                </StackPanel>
+                            </ui:Button.Content>
+                        </ui:Button>
                     </Grid>
                 </StackPanel>
             </ui:Card>

StabilityMatrix/Models/Api/CivitModel.cs

@@ -36,4 +36,27 @@ public class CivitModel
     [BsonRef("ModelVersions")]
     [JsonPropertyName("modelVersions")]
     public List<CivitModelVersion>? ModelVersions { get; set; }
+
+    private FileSizeType? _fullFilesSize;
+    public FileSizeType FullFilesSize
+    {
+        get
+        {
+            if (_fullFilesSize == null)
+            {
+                var kbs = 0.0;
+                if (ModelVersions is not null)
+                {
+                    var latestVersion = ModelVersions[0];
+                    if (latestVersion.Files is not null)
+                    {
+                        var latestModelFile = latestVersion.Files[0];
+                        kbs = latestModelFile.SizeKb;
+                    }
+                }
+                _fullFilesSize = new FileSizeType(kbs);
+            }
+            return _fullFilesSize;
+        }
+    }
 }

StabilityMatrix/Models/FileSizeType.cs

@@ -0,0 +1,36 @@
+using System.Globalization;
+
+namespace StabilityMatrix.Models;
+
+public class FileSizeType
+{
+    public double SizeInKB { get; private set; }
+
+    public string HumanReadableRepresentation { get; private set; }
+
+    public FileSizeType(double sizeInKB)
+    {
+        SizeInKB = sizeInKB;
+        HumanReadableRepresentation = ConvertToHumanReadable();
+    }
+
+    private string ConvertToHumanReadable()
+    {
+        var sizeUnits = new string[] { "KB", "MB", "GB", "TB" };
+        var size = SizeInKB;
+        var unitIndex = 0;
+
+        while (size >= 1024 && unitIndex < sizeUnits.Length - 1)
+        {
+            size /= 1024;
+            unitIndex++;
+        }
+
+        return string.Format("{0} {1}", size.ToString("0.##", CultureInfo.InvariantCulture), sizeUnits[unitIndex]);
+    }
+
+    public override string ToString()
+    {
+        return HumanReadableRepresentation;
+    }
+}