mayf commited on
Commit
a2c88cd
·
verified ·
1 Parent(s): eec20c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.5B",
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},