mirror of
https://github.com/Kuro0911/CS5446-Project.git
synced 2026-02-12 13:02:52 +00:00
prompt based results
This commit is contained in:
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
@@ -0,0 +1,3 @@
|
||||
__pycache__/
|
||||
.ipynb_checkpoints/
|
||||
|
||||
@@ -23,7 +23,7 @@ DEFAULT_MODELS = {
|
||||
|
||||
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
||||
DTYPE = torch.bfloat16 if torch.cuda.is_available() else torch.float32
|
||||
auth_token = "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"
|
||||
auth_token = "HF_TOKEN"
|
||||
|
||||
_PREFERRED_Q4K_ORDER = ("Q4_K_M", "Q4_K_S", "Q4_K_L", "Q4_K")
|
||||
_ENV_LOCAL_GGUF = "HF_GGUF_LOCAL_PATH"
|
||||
|
||||
@@ -70,7 +70,7 @@ def _get_hf_judge():
|
||||
device = 0 if torch.cuda.is_available() else -1
|
||||
dtype = _pick_dtype()
|
||||
|
||||
hf_token = "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"
|
||||
hf_token = "HF_TOKEN"
|
||||
if hf_token is None:
|
||||
raise RuntimeError(
|
||||
"❌ Hugging Face token not found. Set it with:\n"
|
||||
|
||||
2
model.py
2
model.py
@@ -23,7 +23,7 @@ DEFAULT_MODELS = {
|
||||
|
||||
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
||||
DTYPE = torch.bfloat16 if torch.cuda.is_available() else torch.float32
|
||||
auth_token = "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"
|
||||
auth_token = "HF_TOKEN"
|
||||
|
||||
_PREFERRED_Q4K_ORDER = ("Q4_K_M", "Q4_K_S", "Q4_K_L", "Q4_K")
|
||||
_ENV_LOCAL_GGUF = "HF_GGUF_LOCAL_PATH"
|
||||
|
||||
@@ -70,7 +70,7 @@ def _get_hf_judge():
|
||||
device = 0 if torch.cuda.is_available() else -1
|
||||
dtype = _pick_dtype()
|
||||
|
||||
hf_token = "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"
|
||||
hf_token = "HF_TOKEN"
|
||||
if hf_token is None:
|
||||
raise RuntimeError(
|
||||
"❌ Hugging Face token not found. Set it with:\n"
|
||||
|
||||
Reference in New Issue
Block a user