Skip to content

Instantly share code, notes, and snippets.

@AmosLewis
Created September 14, 2022 17:21
Show Gist options
  • Save AmosLewis/f160269f42f13dd27434223371725975 to your computer and use it in GitHub Desktop.
Save AmosLewis/f160269f42f13dd27434223371725975 to your computer and use it in GitHub Desktop.
output compare bloom
➜ SHARK git:(bloom) ✗ python tank/bloom_model.py
Some weights of BloomForSequenceClassification were not initialized from the model checkpoint at bigscience/bloom-560m and are newly initialized: ['score.weight']
You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.
/home/chi/src/ubuntu20/shark/SHARK/shark.venv/lib/python3.10/site-packages/torch/jit/_check.py:181: UserWarning: The TorchScript type system doesn't support instance-level annotations on empty non-base types in `__init__`. Instead, either 1) use a type annotation in the class body, or 2) wrap the type in `torch.jit.Attribute`.
warnings.warn("The TorchScript type system doesn't support "
/home/chi/src/ubuntu20/shark/SHARK/shark.venv/lib/python3.10/site-packages/torch/jit/_trace.py:744: UserWarning: The input to trace is already a ScriptModule, tracing it is a no-op. Returning the object as is.
warnings.warn(
Target triple found:x86_64-linux-gnu
tensor([[ 7.2041, -17.0263]], grad_fn=<IndexBackward0>) tensor([[ 9.8524, -19.8780]])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment