Commit
·
002bec9
1
Parent(s):
20354da
updated requirements
Browse files- pyproject.toml +9 -1
- requirements.txt +0 -2
- setup.cfg +0 -12
- setup.py +0 -3
- src/sdgen/main.py +5 -0
pyproject.toml
CHANGED
|
@@ -1,3 +1,11 @@
|
|
| 1 |
[build-system]
|
| 2 |
-
requires = ["setuptools>=
|
| 3 |
build-backend = "setuptools.build_meta"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
[build-system]
|
| 2 |
+
requires = ["setuptools>=45", "wheel"]
|
| 3 |
build-backend = "setuptools.build_meta"
|
| 4 |
+
|
| 5 |
+
[project]
|
| 6 |
+
name = "sdgen"
|
| 7 |
+
version = "0.1.0"
|
| 8 |
+
requires-python = ">=3.9"
|
| 9 |
+
|
| 10 |
+
[tool.setuptools.packages.find]
|
| 11 |
+
where = ["src"]
|
requirements.txt
CHANGED
|
@@ -1,5 +1,3 @@
|
|
| 1 |
-
-e .
|
| 2 |
-
|
| 3 |
# CORE GPU LIBRARIES (PyTorch - GPU ONLY)
|
| 4 |
# IMPORTANT: Install manually using:
|
| 5 |
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
|
|
|
|
|
|
|
|
|
| 1 |
# CORE GPU LIBRARIES (PyTorch - GPU ONLY)
|
| 2 |
# IMPORTANT: Install manually using:
|
| 3 |
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
setup.cfg
DELETED
|
@@ -1,12 +0,0 @@
|
|
| 1 |
-
[metadata]
|
| 2 |
-
name = sdgen
|
| 3 |
-
version = 0.1.0
|
| 4 |
-
|
| 5 |
-
[options]
|
| 6 |
-
packages = find:
|
| 7 |
-
package_dir =
|
| 8 |
-
=src
|
| 9 |
-
python_requires = >=3.8
|
| 10 |
-
|
| 11 |
-
[options.packages.find]
|
| 12 |
-
where = src
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
setup.py
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
from setuptools import setup
|
| 2 |
-
|
| 3 |
-
setup()
|
|
|
|
|
|
|
|
|
|
|
|
src/sdgen/main.py
CHANGED
|
@@ -6,6 +6,8 @@ sets up the UI, and launches the Gradio interface.
|
|
| 6 |
|
| 7 |
from __future__ import annotations
|
| 8 |
|
|
|
|
|
|
|
| 9 |
import torch
|
| 10 |
from dotenv import load_dotenv
|
| 11 |
|
|
@@ -15,6 +17,9 @@ from sdgen.sd.pipeline import load_pipeline, warmup_pipeline
|
|
| 15 |
from sdgen.ui import build_ui
|
| 16 |
from sdgen.utils.logger import get_logger
|
| 17 |
|
|
|
|
|
|
|
|
|
|
| 18 |
logger = get_logger(__name__)
|
| 19 |
load_dotenv()
|
| 20 |
|
|
|
|
| 6 |
|
| 7 |
from __future__ import annotations
|
| 8 |
|
| 9 |
+
import sys
|
| 10 |
+
import os
|
| 11 |
import torch
|
| 12 |
from dotenv import load_dotenv
|
| 13 |
|
|
|
|
| 17 |
from sdgen.ui import build_ui
|
| 18 |
from sdgen.utils.logger import get_logger
|
| 19 |
|
| 20 |
+
# for HF spaces
|
| 21 |
+
sys.path.append(os.path.abspath("src"))
|
| 22 |
+
|
| 23 |
logger = get_logger(__name__)
|
| 24 |
load_dotenv()
|
| 25 |
|