loading the model using from_pretrained method

This commit is contained in:
Rajat Sen
2025-09-26 22:47:56 +00:00
parent 2e6db886fa
commit c42a740917
2 changed files with 136 additions and 152 deletions
+4
View File
@@ -20,6 +20,10 @@ huggingface_hub = { version = ">=0.23.0", extras = ["cli"] }
safetensors = ">=0.5.3"
torch = { version = ">=2.0.0", extras = ["cuda"] }
[tool.ruff]
line-length = 100
indent-width = 2
[build-system]
requires = ["poetry-core"]