Added download script and updated management command

This commit is contained in:
Viswamedha Nalabotu 2026-02-26 11:38:04 +00:00
parent f69d9f3e1d
commit 29f97b383d
2 changed files with 32 additions and 4 deletions

View file

@ -1,14 +1,17 @@
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from accounts.models import User from django.contrib.auth import get_user_model
User = get_user_model()
class Command(BaseCommand): class Command(BaseCommand):
help = 'Resets non-admin account passwords to "password" and admin account passwords to "admin" using the current SECRET_KEY' help = 'Resets non-admin account passwords to "password" and admin account passwords to "admin"'
def handle(self, *args, **kwargs): def handle(self, *args, **kwargs):
for user in User.objects.all(): users = User.objects.all()
for user in users:
if user.is_staff: if user.is_staff:
user.set_password('admin') user.set_password('admin')
else: else:
user.set_password('password') user.set_password('password')
user.save() user.save()
self.stdout.write("All account passwords synchronized with local SECRET_KEY.") self.stdout.write(self.style.SUCCESS(f"Successfully updated {users.count()} users."))

25
download_model.py Normal file
View file

@ -0,0 +1,25 @@
import os
from huggingface_hub import hf_hub_download
REPO_ID = "Bartowski/Meta-Llama-3.1-8B-Instruct-GGUF"
FILENAME = "Meta-Llama-3.1-8B-Instruct-Q4_K_M.gguf"
LOCAL_DIR = "models"
def download_model():
print(f"🚀 Starting download of {FILENAME}...")
print(f"📂 Destination: {os.path.abspath(LOCAL_DIR)}")
try:
path = hf_hub_download(
repo_id=REPO_ID,
filename=FILENAME,
local_dir=LOCAL_DIR,
)
print(f"Model downloaded to: {path}")
print(f"Expected size: ~4.92 GB")
except Exception as e:
print(f"Error downloading model: {e}")
if __name__ == "__main__":
os.makedirs(LOCAL_DIR, exist_ok=True)
download_model()