pyg-nightly 2.7.0.dev20250922__py3-none-any.whl → 2.7.0.dev20250924__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250922.dist-info → pyg_nightly-2.7.0.dev20250924.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250922
+Version: 2.7.0.dev20250924
 Summary: Graph Neural Network Library for PyTorch
 Keywords: deep-learning,pytorch,geometric-deep-learning,graph-neural-networks,graph-convolutional-networks
 Author-email: Matthias Fey <matthias@pyg.org>

{pyg_nightly-2.7.0.dev20250922.dist-info → pyg_nightly-2.7.0.dev20250924.dist-info}/RECORD

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=CVSPcDYT5FKUlUV-SUiXxZibED2VnZwcdi3iLa9WmCU,2292
+torch_geometric/__init__.py,sha256=ZrtnmLssSCHHVi0DleX_SBMUPOZEpNy-QDFULbz-Idk,2292
 torch_geometric/_compile.py,sha256=9yqMTBKatZPr40WavJz9FjNi7pQj8YZAZOyZmmRGXgc,1351
 torch_geometric/_onnx.py,sha256=ODB_8cwFUiwBUjngXn6-K5HHb7IDul7DDXuuGX7vj_0,8178
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -274,7 +274,7 @@ torch_geometric/llm/models/__init__.py,sha256=Xb2GacStLOYCAu34NNzG-zog-zapXh1asl
 torch_geometric/llm/models/g_retriever.py,sha256=SeW0rlrlzMhN3aVslhq_GUyUgS4sVw_nMAT5YiXzZd8,9072
 torch_geometric/llm/models/git_mol.py,sha256=m1YJb6Xb2i6j9wEqHzqE4YBWMr9i1CJpMp6T-E24fsA,12680
 torch_geometric/llm/models/glem.py,sha256=GlL_I63g-_5eTycSGRj720YntldQ-CQ351RaDPc6XAU,16674
-torch_geometric/llm/models/llm.py,sha256=LKYdUSw2PSWYEN13DYYaJt0dHRQTptX6mM1rAh7-J6c,17764
+torch_geometric/llm/models/llm.py,sha256=039mq9rZBZMyZW5rYj0fMP5kl9RJAI7N-oJyC4Mf2Gs,18122
 torch_geometric/llm/models/llm_judge.py,sha256=qhc8hmIPNhcfLVRyBVk7jQW7ncoIb9QYw7rcsGAIpyg,6457
 torch_geometric/llm/models/molecule_gpt.py,sha256=RWoP4RMsoRzZtuedPCLNCfooqibCqxkuAhH-pyek9No,7641
 torch_geometric/llm/models/protein_mpnn.py,sha256=SwTgafSbI2KJ-yqzn0trZtVWLmfo0_kPEaWSNJUCt70,12266
@@ -654,7 +654,7 @@ torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5
 torch_geometric/visualization/__init__.py,sha256=b-HnVesXjyJ_L1N-DnjiRiRVf7lhwKaBQF_2i5YMVSU,208
 torch_geometric/visualization/graph.py,sha256=mfZHXYfiU-CWMtfawYc80IxVwVmtK9hbIkSKhM_j7oI,14311
 torch_geometric/visualization/influence.py,sha256=CWMvuNA_Nf1sfbJmQgn58yS4OFpeKXeZPe7kEuvkUBw,477
-pyg_nightly-2.7.0.dev20250922.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250922.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250922.dist-info/METADATA,sha256=K_NpLbfJp906gEJXyyYtdzIaMDfj5EX-7su-whiTE5A,63680
-pyg_nightly-2.7.0.dev20250922.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250924.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250924.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250924.dist-info/METADATA,sha256=32NMS3YGNQGFJCeXcyOzlqjzFiwXsx0WLfWHhD2Y3EI,63680
+pyg_nightly-2.7.0.dev20250924.dist-info/RECORD,,

torch_geometric/__init__.py

@@ -31,7 +31,7 @@ from .lazy_loader import LazyLoader
 contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
 graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
 
-__version__ = '2.7.0.dev20250922'
+__version__ = '2.7.0.dev20250924'
 
 __all__ = [
     'Index',

torch_geometric/llm/models/llm.py

@@ -88,8 +88,8 @@ class LLM(torch.nn.Module):
                 num_params = float(list(param_count.values())[0] // 10**9)
 
             # A rough heuristic on GPU memory requirements, e.g., we found that
-            # LLAMA2 (7B parameters) fits on a 85GB GPU.
-            required_memory = 85 * num_params / 7
+            # LLAMA3 (8B parameters) fits on a 96GB GPU.
+            required_memory = 96.0 * num_params / 8.0
             kwargs = get_llm_kwargs(required_memory, dtype)
         else:
             gpu_memory: List[int] = []
@@ -136,8 +136,13 @@ class LLM(torch.nn.Module):
         else:
             self.sys_prompt = ""
         if 'max_memory' not in kwargs:  # Pure CPU:
-            warnings.warn("LLM is being used on CPU, which may be slow",
-                          stacklevel=2)
+            warnings.warn(
+                "LLM is being used on CPU, which may be slow. This decision "
+                "was made by a rough hueristic that assumes your GPU set up "
+                "does not have enough GPU RAM. This is done to avoid GPU OOM "
+                "errors. If you think this is a mistake, please initialize "
+                "your LLM with the n_gpus param to dictate how many gpus to "
+                "use for the LLM.", stacklevel=2)
             self.device = torch.device('cpu')
             self.autocast_context = nullcontext()
         else: