lea97338 commited on
Commit
d407e28
·
verified ·
1 Parent(s): 6fe7eb6

Update app.py

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