Shubham Krishna

Example of Online Clustering

3

This PR aims to illustrates an example of performing online clustering using Stateful Processing and RunInference. The entire implementation is divided into two different pipelines: 1. `write_data_to_pubsub_pipeline` pushes data to...

Shape Error when running on CPU

2

Hi @sanchit-gandhi, thanks for creating this package but the example mentioned by you here: https://github.com/sanchit-gandhi/whisper-jax#model-usage doesn't work for me. I am getting the shape error: ``` ScopeParamShapeError: Initializer expected to...

SIGSEGV: Segmentation Fault Memory error while checkpointing with transformers trainer on a v5-litepod-8 Google Cloud TPU

6

## 🐛 Bug concurrent.futures.process.BrokenProcessPool: A process in the process pool was terminated abruptly while the future was running or pending. root@t1v-n-108b165f-w-0:/workspace# /usr/local/lib/python3.10/multiprocessing/resource_tracker.py:224: UserWarning: resource_tracker: There appear to be 8 leaked...

fix: extend the unwrap_model function and save unwrapped model state dict instead of wrapped

3

# What does this PR do? This PR pushes two changes: - Save the ` unwrap_model(model).state_dict()` whenever`if isinstance(unwrap_model(model), supported_classes)` - Extend the `unwrap_model()` so that any wrapper on the children...