Update README.md
Browse files
README.md
CHANGED
|
@@ -40,6 +40,7 @@ print(tokenizer.decode(model.generate(**inputs, max_new_tokens=50, do_sample=Fal
|
|
| 40 |
```
|
| 41 |
|
| 42 |
### Intel Gaudi-2 INT4 Inference
|
|
|
|
| 43 |
```python
|
| 44 |
import habana_frameworks.torch.core as htcore
|
| 45 |
import habana_frameworks.torch.hpu as hthpu
|
|
|
|
| 40 |
```
|
| 41 |
|
| 42 |
### Intel Gaudi-2 INT4 Inference
|
| 43 |
+
docker image with Gaudi Software Stack is recommended. More details can be found in [Gaudi Guide](https://docs.habana.ai/en/latest/).
|
| 44 |
```python
|
| 45 |
import habana_frameworks.torch.core as htcore
|
| 46 |
import habana_frameworks.torch.hpu as hthpu
|