update

Author: xnuohz

.github/workflows/testing_latest.yml

@@ -69,7 +69,7 @@ jobs:
 
       - name: Run tests
         if: steps.changed-files-specific.outputs.only_changed != 'true'
-        timeout-minutes: 20
+        timeout-minutes: 25
         run: |
           uv run --no-project pytest --cov --cov-report=xml --durations 10
 

examples/llm/molecule_gpt.py

@@ -93,9 +93,10 @@ def get_clippable_params(params):
     # Create model ===============================================
     llm = LLM(
         # model_name='lmsys/vicuna-7b-v1.5',
-        model_name='TinyLlama/TinyLlama-1.1B-Chat-v0.1',
+        model_name='Qwen/Qwen3-0.6B',
         num_params=1,
         dtype=torch.bfloat16,
+        sys_prompt='You are an agent, answer my questions.',
     )
 
     graph_encoder = GINEConv(

torch_geometric/llm/models/llm.py

@@ -43,7 +43,7 @@ def get_llm_kwargs(required_memory: int, dtype=torch.dtype) -> Dict[str, Any]:
         }
         kwargs['low_cpu_mem_usage'] = True
         kwargs['device_map'] = 'auto'
-        kwargs['dtype'] = dtype
+        kwargs['torch_dtype'] = dtype
 
     return kwargs