Spaces:
Sleeping
Sleeping
Commit ·
f08bbe9
1
Parent(s): 8bd478d
nothing
Browse files
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 |
-
|
| 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 = {...}
|