Brice Vandeputte commited on
Commit
b765164
1 Parent(s): ed31d02

accelerate import

Browse files
Files changed (3) hide show
  1. .gitignore +3 -0
  2. app.py +3 -0
  3. requirements.txt +2 -1
.gitignore CHANGED
@@ -1,2 +1,5 @@
1
  .venv/
2
  __pycache__/
 
 
 
 
1
  .venv/
2
  __pycache__/
3
+
4
+ # IntelliJ
5
+ .idea/
app.py CHANGED
@@ -2,6 +2,9 @@ import collections
2
  import heapq
3
  import json
4
  import os
 
 
 
5
  from accelerate.logging import get_logger
6
 
7
  import gradio as gr
 
2
  import heapq
3
  import json
4
  import os
5
+
6
+ # https://huggingface.co/docs/accelerate/main/en/basic_tutorials/troubleshooting#logging
7
+ # https://pypi.org/project/accelerate/
8
  from accelerate.logging import get_logger
9
 
10
  import gradio as gr
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  open_clip_torch
2
  torchvision
3
  torch
4
- gradio
 
 
1
  open_clip_torch
2
  torchvision
3
  torch
4
+ gradio
5
+ accelerate