-
Framework not specified. Using pt to export the model.
====Exporting IR=====
Loading checkpoint shards: 0%| | 0/7 [00:00
-
I am trying to run a Hamiltonian MCMC on a target distribution whose implementation involves a call to tf.gather. The following code:
```python
import tensorflow as tf
import tensorflow_probabili…
-
git version: c3d3cef8d5837
system: `Ubuntu 18.04.6 LTS`
reproduce with: `mlir-opt -one-shot-bufferize a.mlir`
a.mlir:
```
module {
func.func @test_reshape_3d_same_d2d_auto_empty(%arg0…
-
TS INT8 degradation later versions
Hi all, I get a degradation in results after an INT8 quantization with torchscript, after updating my torch_tensorrt, torch and tensorrt versions. I have listed t…
-
# 🐛 ValueError while converting gpytorch model to ONNX
I followed this
[Converting Exact GP Models to TorchScript](https://github.com/cornellius-gp/gpytorch/blob/ff1881b5fe92147f5e34e4141c0d5255e2…
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
No
### Source
binary
### TensorFlow version
2.16.1
### Custom code
Yes
### OS platform and dis…
-
🚀 The feature, motivation and pitch
# RFC: Multi-Gpu Python Frontend API
This RFC compares and contrasts some ideas for exposing multi-gpu support in the python frontend.
1. The current `multigpu_sc…
-
RuntimeWarning: Iterating over a tensor might cause the trace to be incorrect. Passing a tensor of different shape won't change the number of iterations executed (and might lead to errors or silently …
-
```
def test_trace_modern_ctor(self):
class MyModule(nn.Module):
def forward(self, x):
return torch.tensor([0.0])
# Shouldn't fail sanity checks
…
-
Using `.token[0]` returns the padding token
```py
from nnsight import LanguageModel
model = LanguageModel("gpt2", device_map="cpu")
probs = model.trace("a zd zdb", trace=False).logits
with mode…