Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -25,7 +25,6 @@ with init_empty_weights():
|
|
| 25 |
del dummy_model
|
| 26 |
|
| 27 |
for k, v in device_map.items():
|
| 28 |
-
print(f"{k} => {v}")
|
| 29 |
device_map[k] = 'cpu'
|
| 30 |
|
| 31 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 25 |
del dummy_model
|
| 26 |
|
| 27 |
for k, v in device_map.items():
|
|
|
|
| 28 |
device_map[k] = 'cpu'
|
| 29 |
|
| 30 |
model = AutoModelForCausalLM.from_pretrained(
|