Hugging Face's logo

Spaces:
GIGAParviz
/
Firooze_test
Runtime error

App Files Community
GIGAParviz commited on
Commit
c5f8dce
·
verified ·
1 Parent(s): dc9939b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def clear_memory():
11
  model_name = "GIGAParviz/Firooze_test"
12
 
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
- model = AutoModelForCausalLM.from_pretrained(model_name)
15
  model = model.to("cpu")
16
  model.gradient_checkpointing_enable()
17
 
 
11
  model_name = "GIGAParviz/Firooze_test"
12
 
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
+ model = AutoModelForCausalLM.from_pretrained(model_name , low_cpu_mem_usage=True , device_map="cpu")
15
  model = model.to("cpu")
16
  model.gradient_checkpointing_enable()
17