Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ dtype = torch.float32
|
|
| 7 |
|
| 8 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-1.5B")
|
| 9 |
text_encoder = Qwen2ForCausalLM.from_pretrained(
|
| 10 |
-
"Qwen/Qwen2-
|
| 11 |
torch_dtype=dtype,
|
| 12 |
)
|
| 13 |
|
|
|
|
| 7 |
|
| 8 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-1.5B")
|
| 9 |
text_encoder = Qwen2ForCausalLM.from_pretrained(
|
| 10 |
+
"Qwen/Qwen2-3B",
|
| 11 |
torch_dtype=dtype,
|
| 12 |
)
|
| 13 |
|