| import sys, os, json |
| root = os.sep + os.sep.join(__file__.split(os.sep)[1:__file__.split(os.sep).index("Recurrent-Parameter-Generation")+1]) |
| sys.path.append(root) |
| os.chdir(root) |
|
|
|
|
| from workspace.classinput.generate import generate |
| from workspace.classinput.qwen25llm import get_embedding |
| import torch |
| import time |
|
|
|
|
|
|
|
|
| while True: |
| time.sleep(0.5) |
| save_name = "./workspace/classinput/generated_class{}.pth" |
| print("\n\n\n==================================================================================") |
| print('class includes: ("airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse", "ship", "truck")') |
| text_emb = input("Input your description: ") or "Give me a model to select all living things." |
| |
| emb = get_embedding(prompt=text_emb) |
| emb = torch.tensor(emb, dtype=torch.float) |
| params = generate(save_path=save_name, embedding=emb) |
|
|
