-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Description
Linux Mint (Ubuntu-based)
made a fork of the main repo and cloned it into Visual Studio Code
Python 3.12.3 .venv
running coverage run --source=llmrag -m unittest discover -s tests
gets me the error TypeError: TransformersModel.generate() got an unexpected keyword argument 'temperature'
(.venv) fynn@ThinkPad:~/Coding/SC RAG/llmragSC$ coverage run --source=llmrag -m unittest discover -s tests
config.json: 100%|█████████████████████████████████████████████████████████████████████████████| 665/665 [00:00<00:00, 1.22MB/s]
model.safetensors: 100%|█████████████████████████████████████████████████████████████████████| 548M/548M [00:40<00:00, 13.6MB/s]
generation_config.json: 100%|███████████████████████████████████████████████████████████████████| 124/124 [00:00<00:00, 449kB/s]
tokenizer_config.json: 100%|█████████████████████████████████████████████████████████████████| 26.0/26.0 [00:00<00:00, 38.3kB/s]
vocab.json: 100%|██████████████████████████████████████████████████████████████████████████| 1.04M/1.04M [00:00<00:00, 4.25MB/s]
merges.txt: 100%|████████████████████████████████████████████████████████████████████████████| 456k/456k [00:00<00:00, 3.62MB/s]
tokenizer.json: 100%|██████████████████████████████████████████████████████████████████████| 1.36M/1.36M [00:00<00:00, 16.4MB/s]
Device set to use cpu
Failed to send telemetry event ClientStartEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event ClientCreateCollectionEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event CollectionGetEvent: capture() takes 1 positional argument but 3 were given
/home/fynn/.cache/chroma/onnx_models/all-MiniLM-L6-v2/onnx.tar.gz: 100%|██████████████████| 79.3M/79.3M [00:10<00:00, 7.93MiB/s]
Failed to send telemetry event CollectionAddEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event CollectionQueryEvent: capture() takes 1 positional argument but 3 were given
EEFailed to send telemetry event ClientStartEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event ClientCreateCollectionEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event ClientStartEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event ClientCreateCollectionEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event CollectionAddEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event CollectionQueryEvent: capture() takes 1 positional argument but 3 were given
Retrieved: [('Paris is the capital of France.', 0.2878603935241699)]
.Failed to send telemetry event ClientStartEvent: capture() takes 1 positional argument but 3 were given
Failed to send telemetry event ClientCreateCollectionEvent: capture() takes 1 positional argument but 3 were given
Retrieved: [('Paris is the capital of France.', 0.370265930891037)]
.
======================================================================
ERROR: test_query (test_pipeline.TestRAGPipeline.test_query)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/fynn/Coding/SC RAG/llmragSC/tests/test_pipeline.py", line 40, in test_query
answer = self.pipeline.query(question)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/fynn/Coding/SC RAG/llmragSC/llmrag/pipelines/rag_pipeline.py", line 49, in query
return self.model.generate(prompt, temperature=temperature)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: TransformersModel.generate() got an unexpected keyword argument 'temperature'
======================================================================
ERROR: test_retrieve_and_generate (test_pipeline.TestRAGPipeline.test_retrieve_and_generate)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/fynn/Coding/SC RAG/llmragSC/tests/test_pipeline.py", line 34, in test_retrieve_and_generate
response = self.pipeline.query(question)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/fynn/Coding/SC RAG/llmragSC/llmrag/pipelines/rag_pipeline.py", line 49, in query
return self.model.generate(prompt, temperature=temperature)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: TransformersModel.generate() got an unexpected keyword argument 'temperature'
----------------------------------------------------------------------
Ran 5 tests in 76.632s
FAILED (errors=2)Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels