From 551571cb751eedf55891c3ad690182e0d48c7759 Mon Sep 17 00:00:00 2001
From: Alexandru Gherghescu <gherghescu_alex1@yahoo.ro>
Date: Fri, 2 Feb 2024 17:29:30 +0200
Subject: [PATCH] Add 35M model setup
---
scripts/memory_compute_estimations/compute_req.py | 1 +
scripts/memory_compute_estimations/memory_req.py | 1 +
2 files changed, 2 insertions(+)
diff --git a/scripts/memory_compute_estimations/compute_req.py b/scripts/memory_compute_estimations/compute_req.py
index 5f3188a..12e41a8 100644
--- a/scripts/memory_compute_estimations/compute_req.py
+++ b/scripts/memory_compute_estimations/compute_req.py
@@ -1,4 +1,5 @@
setups = {
+ "35M": { "L": 6, "H": 8, "D": 512, },
"70M": { "L": 10, "H": 10, "D": 640, },
"284M": { "L": 20, "H": 16, "D": 1024, },
"512M": { "L": 24, "H": 10, "D": 1280, },
diff --git a/scripts/memory_compute_estimations/memory_req.py b/scripts/memory_compute_estimations/memory_req.py
index 0edcbd4..3439159 100644
--- a/scripts/memory_compute_estimations/memory_req.py
+++ b/scripts/memory_compute_estimations/memory_req.py
@@ -1,4 +1,5 @@
setups = {
+ "35M": { "L": 6, "H": 8, "D": 512, },
"70M": { "L": 10, "H": 10, "D": 640, },
"284M": { "L": 20, "H": 16, "D": 1024, },
"512M": { "L": 24, "H": 10, "D": 1280, },
--
GitLab