Skip to content

Commit 66d6447

Browse files
committed
update repository location
1 parent 21a7253 commit 66d6447

File tree

13 files changed

+109
-67
lines changed

13 files changed

+109
-67
lines changed

ModelForge/routers/models_router.py

Lines changed: 46 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,52 @@
33
Handles model listing and retrieval operations.
44
"""
55
from fastapi import APIRouter, HTTPException, Depends
6-
from typing import List, Dict
6+
from typing import List, Dict, Any
7+
from pydantic import BaseModel
78

89
from ..services.model_service import ModelService
910
from ..dependencies import get_model_service
1011
from ..logging_config import logger
1112

1213

14+
class ModelsResponse(BaseModel):
15+
"""Response model for list of models."""
16+
models: List[Dict[str, Any]]
17+
18+
1319
router = APIRouter(prefix="/models")
1420

1521

16-
@router.get("/", response_model=List[Dict])
22+
@router.get("/all", response_model=ModelsResponse)
23+
async def get_all_models_legacy(
24+
model_service: ModelService = Depends(get_model_service),
25+
):
26+
"""
27+
Get all fine-tuned models (legacy endpoint for backward compatibility).
28+
29+
Args:
30+
model_service: Model service instance
31+
32+
Returns:
33+
Dictionary with models list
34+
"""
35+
logger.info("Fetching all models (legacy endpoint)")
36+
try:
37+
models = model_service.get_all_models()
38+
# Transform field names to match frontend expectations
39+
transformed_models = []
40+
for model in models:
41+
transformed_model = model.copy()
42+
transformed_model["model_name"] = model["name"]
43+
transformed_model["model_path"] = model["path"]
44+
transformed_models.append(transformed_model)
45+
return ModelsResponse(models=transformed_models)
46+
except Exception as e:
47+
logger.error(f"Error fetching models: {e}")
48+
raise HTTPException(status_code=500, detail=str(e))
49+
50+
51+
@router.get("/", response_model=ModelsResponse)
1752
async def get_all_models(
1853
model_service: ModelService = Depends(get_model_service),
1954
):
@@ -24,12 +59,19 @@ async def get_all_models(
2459
model_service: Model service instance
2560
2661
Returns:
27-
List of all models
62+
Dictionary with models list
2863
"""
2964
logger.info("Fetching all models")
3065
try:
3166
models = model_service.get_all_models()
32-
return models
67+
# Transform field names to match frontend expectations
68+
transformed_models = []
69+
for model in models:
70+
transformed_model = model.copy()
71+
transformed_model["model_name"] = model["name"]
72+
transformed_model["model_path"] = model["path"]
73+
transformed_models.append(transformed_model)
74+
return ModelsResponse(models=transformed_models)
3375
except Exception as e:
3476
logger.error(f"Error fetching models: {e}")
3577
raise HTTPException(status_code=500, detail=str(e))

README.md

Lines changed: 41 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
[![PyPI Downloads](https://static.pepy.tech/personalized-badge/modelforge-finetuning?period=total&units=INTERNATIONAL_SYSTEM&left_color=BLACK&right_color=BLUE&left_text=downloads)](https://pepy.tech/projects/modelforge-finetuning)
44
[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
55
[![Python 3.11](https://img.shields.io/badge/python-3.11-blue.svg)](https://www.python.org/downloads/)
6-
[![Version](https://img.shields.io/badge/version-2.0.1-blue)](https://github.com/RETR0-OS/ModelForge)
6+
[![Version](https://img.shields.io/badge/version-2.0.1-blue)](https://github.com/ForgeOpus/ModelForge)
77

88
**Fine-tune LLMs on your laptop's GPU—no code, no PhD, no hassle.**
99

@@ -20,7 +20,7 @@ ModelForge v2.0 is a complete architectural overhaul bringing **2x faster traini
2020
- 🏗️ **Modular Architecture** for easy extensibility
2121
- 🔒 **Production-Ready** with proper error handling and logging
2222

23-
**[See What's New in v2.0 →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/getting-started/whats-new.md)**
23+
**[See What's New in v2.0 →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/getting-started/whats-new.md)**
2424

2525
## 🚀 Features
2626

@@ -50,7 +50,7 @@ ModelForge v2.0 is a complete architectural overhaul bringing **2x faster traini
5050

5151
> **⚠️ macOS is NOT supported.** ModelForge requires NVIDIA CUDA which is not available on macOS. Use Linux or Windows with NVIDIA GPU.
5252
>
53-
> **Windows Users**: See [Windows Installation Guide](https://github.com/RETR0-OS/ModelForge/tree/main/docs/installation/windows.md) for platform-specific instructions, especially for Unsloth support.
53+
> **Windows Users**: See [Windows Installation Guide](https://github.com/ForgeOpus/ModelForge/tree/main/docs/installation/windows.md) for platform-specific instructions, especially for Unsloth support.
5454
5555
### Installation
5656

@@ -88,52 +88,52 @@ modelforge run
8888

8989
Open your browser to **http://localhost:8000** and start training!
9090

91-
**[Full Quick Start Guide →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/getting-started/quickstart.md)**
91+
**[Full Quick Start Guide →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/getting-started/quickstart.md)**
9292

9393
## 📚 Documentation
9494

9595
### Getting Started
96-
- **[Quick Start Guide](https://github.com/RETR0-OS/ModelForge/tree/main/docs/getting-started/quickstart.md)** - Get up and running in 5 minutes
97-
- **[What's New in v2.0](https://github.com/RETR0-OS/ModelForge/tree/main/docs/getting-started/whats-new.md)** - Major features and improvements
96+
- **[Quick Start Guide](https://github.com/ForgeOpus/ModelForge/tree/main/docs/getting-started/quickstart.md)** - Get up and running in 5 minutes
97+
- **[What's New in v2.0](https://github.com/ForgeOpus/ModelForge/tree/main/docs/getting-started/whats-new.md)** - Major features and improvements
9898

9999
### Installation
100-
- **[Windows Installation](https://github.com/RETR0-OS/ModelForge/tree/main/docs/installation/windows.md)** - Complete Windows setup (including WSL and Docker)
101-
- **[Linux Installation](https://github.com/RETR0-OS/ModelForge/tree/main/docs/installation/linux.md)** - Linux setup guide
102-
- **[Post-Installation](https://github.com/RETR0-OS/ModelForge/tree/main/docs/installation/post-installation.md)** - Initial configuration
100+
- **[Windows Installation](https://github.com/ForgeOpus/ModelForge/tree/main/docs/installation/windows.md)** - Complete Windows setup (including WSL and Docker)
101+
- **[Linux Installation](https://github.com/ForgeOpus/ModelForge/tree/main/docs/installation/linux.md)** - Linux setup guide
102+
- **[Post-Installation](https://github.com/ForgeOpus/ModelForge/tree/main/docs/installation/post-installation.md)** - Initial configuration
103103

104104
### Configuration & Usage
105-
- **[Configuration Guide](https://github.com/RETR0-OS/ModelForge/tree/main/docs/configuration/configuration-guide.md)** - All configuration options
106-
- **[Dataset Formats](https://github.com/RETR0-OS/ModelForge/tree/main/docs/configuration/dataset-formats.md)** - Preparing your training data
107-
- **[Training Tasks](https://github.com/RETR0-OS/ModelForge/tree/main/docs/configuration/training-tasks.md)** - Understanding different tasks
108-
- **[Hardware Profiles](https://github.com/RETR0-OS/ModelForge/tree/main/docs/configuration/hardware-profiles.md)** - Optimizing for your GPU
105+
- **[Configuration Guide](https://github.com/ForgeOpus/ModelForge/tree/main/docs/configuration/configuration-guide.md)** - All configuration options
106+
- **[Dataset Formats](https://github.com/ForgeOpus/ModelForge/tree/main/docs/configuration/dataset-formats.md)** - Preparing your training data
107+
- **[Training Tasks](https://github.com/ForgeOpus/ModelForge/tree/main/docs/configuration/training-tasks.md)** - Understanding different tasks
108+
- **[Hardware Profiles](https://github.com/ForgeOpus/ModelForge/tree/main/docs/configuration/hardware-profiles.md)** - Optimizing for your GPU
109109

110110
### Providers
111-
- **[Provider Overview](https://github.com/RETR0-OS/ModelForge/tree/main/docs/providers/overview.md)** - Understanding providers
112-
- **[HuggingFace Provider](https://github.com/RETR0-OS/ModelForge/tree/main/docs/providers/huggingface.md)** - Standard HuggingFace models
113-
- **[Unsloth Provider](https://github.com/RETR0-OS/ModelForge/tree/main/docs/providers/unsloth.md)** - 2x faster training
111+
- **[Provider Overview](https://github.com/ForgeOpus/ModelForge/tree/main/docs/providers/overview.md)** - Understanding providers
112+
- **[HuggingFace Provider](https://github.com/ForgeOpus/ModelForge/tree/main/docs/providers/huggingface.md)** - Standard HuggingFace models
113+
- **[Unsloth Provider](https://github.com/ForgeOpus/ModelForge/tree/main/docs/providers/unsloth.md)** - 2x faster training
114114

115115
### Training Strategies
116-
- **[Strategy Overview](https://github.com/RETR0-OS/ModelForge/tree/main/docs/strategies/overview.md)** - Understanding strategies
117-
- **[SFT Strategy](https://github.com/RETR0-OS/ModelForge/tree/main/docs/strategies/sft.md)** - Standard supervised fine-tuning
118-
- **[QLoRA Strategy](https://github.com/RETR0-OS/ModelForge/tree/main/docs/strategies/qlora.md)** - Memory-efficient training
119-
- **[RLHF Strategy](https://github.com/RETR0-OS/ModelForge/tree/main/docs/strategies/rlhf.md)** - Reinforcement learning
120-
- **[DPO Strategy](https://github.com/RETR0-OS/ModelForge/tree/main/docs/strategies/dpo.md)** - Direct preference optimization
116+
- **[Strategy Overview](https://github.com/ForgeOpus/ModelForge/tree/main/docs/strategies/overview.md)** - Understanding strategies
117+
- **[SFT Strategy](https://github.com/ForgeOpus/ModelForge/tree/main/docs/strategies/sft.md)** - Standard supervised fine-tuning
118+
- **[QLoRA Strategy](https://github.com/ForgeOpus/ModelForge/tree/main/docs/strategies/qlora.md)** - Memory-efficient training
119+
- **[RLHF Strategy](https://github.com/ForgeOpus/ModelForge/tree/main/docs/strategies/rlhf.md)** - Reinforcement learning
120+
- **[DPO Strategy](https://github.com/ForgeOpus/ModelForge/tree/main/docs/strategies/dpo.md)** - Direct preference optimization
121121

122122
### API Reference
123-
- **[REST API](https://github.com/RETR0-OS/ModelForge/tree/main/docs/api-reference/rest-api.md)** - Complete API documentation
124-
- **[Training Config Schema](https://github.com/RETR0-OS/ModelForge/tree/main/docs/api-reference/training-config.md)** - Configuration options
123+
- **[REST API](https://github.com/ForgeOpus/ModelForge/tree/main/docs/api-reference/rest-api.md)** - Complete API documentation
124+
- **[Training Config Schema](https://github.com/ForgeOpus/ModelForge/tree/main/docs/api-reference/training-config.md)** - Configuration options
125125

126126
### Troubleshooting
127-
- **[Common Issues](https://github.com/RETR0-OS/ModelForge/tree/main/docs/troubleshooting/common-issues.md)** - Frequently encountered problems
128-
- **[Windows Issues](https://github.com/RETR0-OS/ModelForge/tree/main/docs/troubleshooting/windows-issues.md)** - Windows-specific troubleshooting
129-
- **[FAQ](https://github.com/RETR0-OS/ModelForge/tree/main/docs/troubleshooting/faq.md)** - Frequently asked questions
127+
- **[Common Issues](https://github.com/ForgeOpus/ModelForge/tree/main/docs/troubleshooting/common-issues.md)** - Frequently encountered problems
128+
- **[Windows Issues](https://github.com/ForgeOpus/ModelForge/tree/main/docs/troubleshooting/windows-issues.md)** - Windows-specific troubleshooting
129+
- **[FAQ](https://github.com/ForgeOpus/ModelForge/tree/main/docs/troubleshooting/faq.md)** - Frequently asked questions
130130

131131
### Contributing
132-
- **[Contributing Guide](https://github.com/RETR0-OS/ModelForge/tree/main/docs/contributing/contributing.md)** - How to contribute
133-
- **[Architecture](https://github.com/RETR0-OS/ModelForge/tree/main/docs/contributing/architecture.md)** - Understanding the codebase
134-
- **[Model Configurations](https://github.com/RETR0-OS/ModelForge/tree/main/docs/contributing/model-configs.md)** - Adding model recommendations
132+
- **[Contributing Guide](https://github.com/ForgeOpus/ModelForge/tree/main/docs/contributing/contributing.md)** - How to contribute
133+
- **[Architecture](https://github.com/ForgeOpus/ModelForge/tree/main/docs/contributing/architecture.md)** - Understanding the codebase
134+
- **[Model Configurations](https://github.com/ForgeOpus/ModelForge/tree/main/docs/contributing/model-configs.md)** - Adding model recommendations
135135

136-
**[📖 Full Documentation Index →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/README.md)**
136+
**[📖 Full Documentation Index →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/README.md)**
137137

138138
## 🔧 Platform Support
139139

@@ -144,16 +144,16 @@ Open your browser to **http://localhost:8000** and start training!
144144
| **WSL 2** | ✅ Full support | ✅ Full support | Recommended for Windows users |
145145
| **Docker** | ✅ Full support | ✅ Full support | With NVIDIA runtime |
146146

147-
**[Platform-Specific Installation Guides →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/installation/)**
147+
**[Platform-Specific Installation Guides →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/installation/)**
148148

149149
## ⚠️ Important Notes
150150

151151
### Windows Users
152152

153153
**Unsloth provider is NOT supported on native Windows.** For 2x faster training with Unsloth:
154154

155-
1. **Option 1: WSL (Recommended)** - [WSL Installation Guide](https://github.com/RETR0-OS/ModelForge/tree/main/docs/installation/windows.md#option-2-wsl-installation-recommended)
156-
2. **Option 2: Docker** - [Docker Installation Guide](https://github.com/RETR0-OS/ModelForge/tree/main/docs/installation/windows.md#option-3-docker-installation)
155+
1. **Option 1: WSL (Recommended)** - [WSL Installation Guide](https://github.com/ForgeOpus/ModelForge/tree/main/docs/installation/windows.md#option-2-wsl-installation-recommended)
156+
2. **Option 2: Docker** - [Docker Installation Guide](https://github.com/ForgeOpus/ModelForge/tree/main/docs/installation/windows.md#option-3-docker-installation)
157157

158158
The HuggingFace provider works perfectly on native Windows.
159159

@@ -170,7 +170,7 @@ When using Unsloth provider, you **MUST** specify a fixed `max_sequence_length`:
170170

171171
Auto-inference (`max_seq_length: -1`) is **NOT supported** with Unsloth.
172172

173-
**[Learn more about Unsloth →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/providers/unsloth.md)**
173+
**[Learn more about Unsloth →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/providers/unsloth.md)**
174174

175175
## 📂 Dataset Format
176176

@@ -192,7 +192,7 @@ ModelForge uses JSONL format. Each task has specific fields:
192192
{"context": "Document text...", "question": "What is X?", "answer": "X is..."}
193193
```
194194

195-
**[Complete Dataset Format Guide →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/configuration/dataset-formats.md)**
195+
**[Complete Dataset Format Guide →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/configuration/dataset-formats.md)**
196196

197197
## 🤝 Contributing
198198

@@ -204,11 +204,11 @@ We welcome contributions! ModelForge v2.0's modular architecture makes it easy t
204204
- **Improve documentation**
205205
- **Fix bugs and add features**
206206

207-
**[Contributing Guide →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/contributing/contributing.md)**
207+
**[Contributing Guide →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/contributing/contributing.md)**
208208

209209
### Adding Model Recommendations
210210

211-
ModelForge uses modular configuration files for model recommendations. See the **[Model Configuration Guide](https://github.com/RETR0-OS/ModelForge/tree/main/docs/contributing/model-configs.md)** for instructions on adding new recommended models.
211+
ModelForge uses modular configuration files for model recommendations. See the **[Model Configuration Guide](https://github.com/ForgeOpus/ModelForge/tree/main/docs/contributing/model-configs.md)** for instructions on adding new recommended models.
212212

213213
## 🛠 Tech Stack
214214

@@ -232,13 +232,13 @@ MIT License - see [LICENSE](LICENSE) file for details.
232232

233233
## 📧 Support
234234

235-
- **Documentation**: [https://github.com/RETR0-OS/ModelForge/tree/main/docs/](https://github.com/RETR0-OS/ModelForge/tree/main/docs/)
236-
- **Issues**: [GitHub Issues](https://github.com/RETR0-OS/ModelForge/issues)
237-
- **Discussions**: [GitHub Discussions](https://github.com/RETR0-OS/ModelForge/discussions)
235+
- **Documentation**: [https://github.com/ForgeOpus/ModelForge/tree/main/docs/](https://github.com/ForgeOpus/ModelForge/tree/main/docs/)
236+
- **Issues**: [GitHub Issues](https://github.com/ForgeOpus/ModelForge/issues)
237+
- **Discussions**: [GitHub Discussions](https://github.com/ForgeOpus/ModelForge/discussions)
238238
- **PyPI**: [modelforge-finetuning](https://pypi.org/project/modelforge-finetuning/)
239239

240240
---
241241

242242
**ModelForge v2.0 - Making LLM fine-tuning accessible to everyone** 🚀
243243

244-
**[Get Started →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/getting-started/quickstart.md)** | **[Documentation →](https://github.com/RETR0-OS/ModelForge/tree/main/docs/)** | **[GitHub →](https://github.com/RETR0-OS/ModelForge)**
244+
**[Get Started →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/getting-started/quickstart.md)** | **[Documentation →](https://github.com/ForgeOpus/ModelForge/tree/main/docs/)** | **[GitHub →](https://github.com/ForgeOpus/ModelForge)**

docs/README.md

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -62,17 +62,17 @@ Welcome to the comprehensive documentation for **ModelForge v2.0** - A no-code t
6262

6363
## 📖 External Resources
6464

65-
- [GitHub Repository](https://github.com/RETR0-OS/ModelForge)
65+
- [GitHub Repository](https://github.com/ForgeOpus/ModelForge)
6666
- [PyPI Package](https://pypi.org/project/modelforge-finetuning/)
67-
- [Issue Tracker](https://github.com/RETR0-OS/ModelForge/issues)
68-
- [Discussions](https://github.com/RETR0-OS/ModelForge/discussions)
67+
- [Issue Tracker](https://github.com/ForgeOpus/ModelForge/issues)
68+
- [Discussions](https://github.com/ForgeOpus/ModelForge/discussions)
6969

7070
## 💡 Need Help?
7171

7272
- Check the [FAQ](troubleshooting/faq.md) for quick answers
73-
- Search [existing issues](https://github.com/RETR0-OS/ModelForge/issues) on GitHub
74-
- Ask a question in [Discussions](https://github.com/RETR0-OS/ModelForge/discussions)
75-
- Report bugs via [GitHub Issues](https://github.com/RETR0-OS/ModelForge/issues/new)
73+
- Search [existing issues](https://github.com/ForgeOpus/ModelForge/issues) on GitHub
74+
- Ask a question in [Discussions](https://github.com/ForgeOpus/ModelForge/discussions)
75+
- Report bugs via [GitHub Issues](https://github.com/ForgeOpus/ModelForge/issues/new)
7676

7777
---
7878

docs/contributing/contributing.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ Contributors are recognized in:
356356
## Questions?
357357

358358
- Check [FAQ](../troubleshooting/faq.md)
359-
- Ask in [GitHub Discussions](https://github.com/RETR0-OS/ModelForge/discussions)
359+
- Ask in [GitHub Discussions](https://github.com/ForgeOpus/ModelForge/discussions)
360360
- Read [Architecture Guide](architecture.md)
361361

362362
---

docs/contributing/development.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ git clone https://github.com/YOUR-USERNAME/ModelForge.git
3636
cd ModelForge
3737

3838
# Add upstream remote
39-
git remote add upstream https://github.com/RETR0-OS/ModelForge.git
39+
git remote add upstream https://github.com/ForgeOpus/ModelForge.git
4040
```
4141

4242
### 2. Create Virtual Environment

docs/contributing/model-configs.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -415,7 +415,7 @@ Check:
415415
## Questions?
416416

417417
- See [Contributing Guide](contributing.md)
418-
- Ask in [GitHub Discussions](https://github.com/RETR0-OS/ModelForge/discussions)
418+
- Ask in [GitHub Discussions](https://github.com/ForgeOpus/ModelForge/discussions)
419419
- Check [FAQ](../troubleshooting/faq.md)
420420

421421
---

docs/getting-started/quickstart.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ Your previous models and settings are preserved!
150150

151151
- Check the [FAQ](../troubleshooting/faq.md) for common questions
152152
- See [Troubleshooting](../troubleshooting/common-issues.md) for common issues
153-
- Ask in [GitHub Discussions](https://github.com/RETR0-OS/ModelForge/discussions)
153+
- Ask in [GitHub Discussions](https://github.com/ForgeOpus/ModelForge/discussions)
154154

155155
---
156156

docs/installation/linux.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -417,4 +417,4 @@ docker run --gpus all -p 8000:8000 \
417417

418418
---
419419

420-
**Need Help?** Check [Common Issues](../troubleshooting/common-issues.md) or ask in [GitHub Discussions](https://github.com/RETR0-OS/ModelForge/discussions).
420+
**Need Help?** Check [Common Issues](../troubleshooting/common-issues.md) or ask in [GitHub Discussions](https://github.com/ForgeOpus/ModelForge/discussions).

docs/troubleshooting/common-issues.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -177,9 +177,9 @@ pip install unsloth
177177

178178
- [Windows-Specific Issues](windows-issues.md)
179179
- [FAQ](faq.md)
180-
- [GitHub Issues](https://github.com/RETR0-OS/ModelForge/issues)
181-
- [GitHub Discussions](https://github.com/RETR0-OS/ModelForge/discussions)
180+
- [GitHub Issues](https://github.com/ForgeOpus/ModelForge/issues)
181+
- [GitHub Discussions](https://github.com/ForgeOpus/ModelForge/discussions)
182182

183183
---
184184

185-
**Still having issues?** Create an issue on [GitHub](https://github.com/RETR0-OS/ModelForge/issues/new).
185+
**Still having issues?** Create an issue on [GitHub](https://github.com/ForgeOpus/ModelForge/issues/new).

0 commit comments

Comments
 (0)