Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def load_models():
|
|
31 |
# Optimized story generation model
|
32 |
storyteller = pipeline(
|
33 |
"text-generation",
|
34 |
-
model="Qwen/Qwen3-0.
|
35 |
device_map="auto",
|
36 |
trust_remote_code=True,
|
37 |
model_kwargs={"load_in_8bit": True},
|
|
|
31 |
# Optimized story generation model
|
32 |
storyteller = pipeline(
|
33 |
"text-generation",
|
34 |
+
model="Qwen/Qwen3-0.6B",
|
35 |
device_map="auto",
|
36 |
trust_remote_code=True,
|
37 |
model_kwargs={"load_in_8bit": True},
|