GIGAParviz commited on
Commit
dc9939b
·
verified ·
1 Parent(s): 76b6b6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -10,10 +10,8 @@ def clear_memory():
10
 
11
  model_name = "GIGAParviz/Firooze_test"
12
 
13
-
14
- bnb_config = BitsAndBytesConfig(llm_int8_threshold=6.0)
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
- model = AutoModelForCausalLM.from_pretrained(model_name, quantization_config=bnb_config)
17
  model = model.to("cpu")
18
  model.gradient_checkpointing_enable()
19
 
 
10
 
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