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