haxerwddle commited on
Commit
f08bbe9
·
1 Parent(s): 8bd478d
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -48,7 +48,7 @@ tiny_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
48
  tokenizer = AutoTokenizer.from_pretrained(tiny_model)
49
  chat_model = AutoModelForCausalLM.from_pretrained(
50
  tiny_model,
51
- torch_dtype=torch.bfloat16,
52
  device_map="auto",
53
  low_cpu_mem_usage=True
54
  )
@@ -110,7 +110,6 @@ def explain_recycling(class_label):
110
 
111
  # =========================
112
  # WASTE ANALYSIS
113
- # (your 3 versions)
114
  # =========================
115
  waste_analyzation = {...} # KEEP SAME CONTENTS
116
  waste_analyzation_v2 = {...}
 
48
  tokenizer = AutoTokenizer.from_pretrained(tiny_model)
49
  chat_model = AutoModelForCausalLM.from_pretrained(
50
  tiny_model,
51
+ dtype=torch.bfloat16,
52
  device_map="auto",
53
  low_cpu_mem_usage=True
54
  )
 
110
 
111
  # =========================
112
  # WASTE ANALYSIS
 
113
  # =========================
114
  waste_analyzation = {...} # KEEP SAME CONTENTS
115
  waste_analyzation_v2 = {...}