Hugging Face's logo

Spaces:
tvosch
/
VRAM-estimator
Running

App Files Community
tvosch commited on
Commit
c34077d
•
1 Parent(s): 20e8864

new emoji 🧮

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +1 -1
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  title: VRAM Estimator
3
- emoji: 🏢
4
  colorFrom: yellow
5
  colorTo: blue
6
  sdk: gradio
 
1
  ---
2
  title: VRAM Estimator
3
+ emoji: 🧮
4
  colorFrom: yellow
5
  colorTo: blue
6
  sdk: gradio
app.py CHANGED
@@ -16,7 +16,7 @@ PRECISION_TO_BYTES = {"float32": 4,
16
  "int8": 1}
17
 
18
  ZERO_STAGES = [0, 1, 2, 3]
19
- BATCH_SIZES = [1,2,4,8,16,32,64]
20
  OPTIMIZERS = ["adam", "adamw", "sgd"]
21
  HUGGINGFACE_URL_CONFIG = "https://huggingface.co/{}/resolve/main/config.json"
22
 
 
16
  "int8": 1}
17
 
18
  ZERO_STAGES = [0, 1, 2, 3]
19
+ BATCH_SIZES = [1, 2, 4, 8, 16, 32, 64]
20
  OPTIMIZERS = ["adam", "adamw", "sgd"]
21
  HUGGINGFACE_URL_CONFIG = "https://huggingface.co/{}/resolve/main/config.json"
22