mirror of
https://github.com/wassname/discovering_latent_knowledge.git
synced 2026-09-11 12:10:11 +08:00
still not working
This commit is contained in:
@@ -179,3 +179,7 @@ stylaised knowledge:
|
||||
|
||||
|
||||
How to get the prompt? more direct. Just a lying one. Just a true one.
|
||||
|
||||
|
||||
Maybe just try:
|
||||
"The following movie review expresses what sentiment?" just like in ELK and lillian wangs...
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -4,6 +4,8 @@ os.environ['CUDA_VISIBLE_DEVICES']="-1"
|
||||
import torch
|
||||
import argparse
|
||||
from transformers import AutoTokenizer, AutoModelForCausalLM
|
||||
from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE
|
||||
from pathlib import Path
|
||||
|
||||
model_options = dict(
|
||||
device_map="auto",
|
||||
@@ -31,6 +33,13 @@ def main(model_repo, lora_repo = None, **download_options):
|
||||
|
||||
|
||||
if __name__=="__main__":
|
||||
|
||||
files = [f.relative_to(HUGGINGFACE_HUB_CACHE) for f in Path(HUGGINGFACE_HUB_CACHE).glob('models--*')]
|
||||
files = "\n".join(sorted([str(f).replace('--', '/') for f in files]))
|
||||
print(HUGGINGFACE_HUB_CACHE)
|
||||
print("Downloaded models:\n", files)
|
||||
1/0
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('model_repo', type=str)
|
||||
parser.add_argument('-l', '--lora_repo', type=str, default=None, help='Name of the lora repo')
|
||||
|
||||
Reference in New Issue
Block a user