POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod allegro

This commit is contained in:
l.gabrysiak 2025-02-28 21:28:33 +01:00
parent 12cef050a2
commit b895fda3b0
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
allegro.py
View File

@ -1,11 +1,12 @@
import os import os
os.environ["TOKENIZERS_PARALLELISM"] = "false" os.environ["TOKENIZERS_PARALLELISM"] = "false"
import torch
import faiss
from weaviate.connect import ConnectionParams from weaviate.connect import ConnectionParams
import weaviate import weaviate
import torch
import faiss
import numpy as np
from sentence_transformers import SentenceTransformer from sentence_transformers import SentenceTransformer
from datasets import Dataset from datasets import Dataset
from peft import LoraConfig, get_peft_model from peft import LoraConfig, get_peft_model