| | 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) |
| |
|
