Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import torch
|
|
| 5 |
|
| 6 |
# Use a CPU-compatible base model (replace this with your actual full-precision model)
|
| 7 |
base_model_id = "unsloth/gemma-2b" # Replace with real CPU-compatible model
|
| 8 |
-
lora_model_id = "Futuresony/
|
| 9 |
|
| 10 |
# Load the base model on CPU
|
| 11 |
base_model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 5 |
|
| 6 |
# Use a CPU-compatible base model (replace this with your actual full-precision model)
|
| 7 |
base_model_id = "unsloth/gemma-2b" # Replace with real CPU-compatible model
|
| 8 |
+
lora_model_id = "Futuresony/CCM-AI"
|
| 9 |
|
| 10 |
# Load the base model on CPU
|
| 11 |
base_model = AutoModelForCausalLM.from_pretrained(
|