How to delete downloaded model from google colab?

#18
by kickb - opened

I tried using this model in google colab. This is the code:

from langchain.llms import HuggingFacePipeline
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline

model_id="WizardLM/WizardCoder-Python-34B-V1.0"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)
pipe = pipeline(
"text-generation", model=model, tokenizer=tokenizer, max_new_tokens=10
)
hf = HuggingFacePipeline(pipeline=pipe

It downloaded about 70-80 GB of data and stopped executing the code inbetween as it ran out of RAM.
How can I delete all the downloaded data from my google colab environment?

WizardLM changed discussion status to closed

Sign up or log in to comment