From a1650bba072bf8150efeaebcdbf39a094148a1e4 Mon Sep 17 00:00:00 2001 From: BuildTools Date: Sat, 17 Aug 2024 13:27:18 -0700 Subject: [PATCH] chore: add missing requirements - add missing requirements - add dotenv file loading --- requirements.txt | 6 ++++-- src/AutoGGUF.py | 3 +++ 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 3587e22..46df8b2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,11 @@ psutil~=6.0.0 requests~=2.32.3 numpy<2.0.0 -torch~=2.4.0 +torch~=1.13.1 sentencepiece~=0.2.0 PyYAML~=6.0.2 pynvml~=11.5.3 PySide6~=6.7.2 -flask~=3.0.3 \ No newline at end of file +flask~=3.0.3 +python-dotenv~=1.0.1 +safetensors~=0.4.4 \ No newline at end of file diff --git a/src/AutoGGUF.py b/src/AutoGGUF.py index 77dde6e..b521efa 100644 --- a/src/AutoGGUF.py +++ b/src/AutoGGUF.py @@ -4,6 +4,7 @@ from functools import partial from datetime import datetime +from dotenv import load_dotenv from PySide6.QtCore import * from PySide6.QtGui import * from PySide6.QtWidgets import * @@ -42,6 +43,8 @@ def __init__(self): self.setGeometry(100, 100, width, height) self.setWindowFlag(Qt.FramelessWindowHint) + load_dotenv() # Loads the .env file + self.resize_factor = 1.1 # 10% increase/decrease self.default_width, self.default_height = self.parse_resolution() self.resize(self.default_width, self.default_height)