rahul7star commited on
Commit
1d1ff85
Β·
verified Β·
1 Parent(s): d953fb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -9,14 +9,14 @@ from transformers import AutoProcessor, Gemma3ForConditionalGeneration, TextIter
9
  import time
10
 
11
  # model config
12
- model_12b_name = "rahul7star/gemma-3bit"
13
  model_4b_name = "rahul7star/gemma-3bit"
14
- model_12b = Gemma3ForConditionalGeneration.from_pretrained(
15
- model_12b_name,
16
- device_map="auto",
17
- torch_dtype=torch.bfloat16
18
- ).eval()
19
- processor_12b = AutoProcessor.from_pretrained(model_12b_name)
20
  model_4b = Gemma3ForConditionalGeneration.from_pretrained(
21
  model_4b_name,
22
  device_map="auto",
 
9
  import time
10
 
11
  # model config
12
+ #model_12b_name = "rahul7star/gemma-3bit"
13
  model_4b_name = "rahul7star/gemma-3bit"
14
+ # model_12b = Gemma3ForConditionalGeneration.from_pretrained(
15
+ # model_12b_name,
16
+ # device_map="auto",
17
+ # torch_dtype=torch.bfloat16
18
+ # ).eval()
19
+ #processor_12b = AutoProcessor.from_pretrained(model_12b_name)
20
  model_4b = Gemma3ForConditionalGeneration.from_pretrained(
21
  model_4b_name,
22
  device_map="auto",