Azure99 commited on
Commit
ddaba6e
1 Parent(s): bb9558e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
10
 
11
  device = torch.device("cuda:0")
12
 
13
- llm = AutoModelForCausalLM.from_pretrained("Azure99/blossom-v5-9b", torch_dtype=torch.float16, device_map="auto")
14
- tokenizer = AutoTokenizer.from_pretrained("Azure99/blossom-v5-9b")
15
  diffusion_pipe = DiffusionPipeline.from_pretrained(
16
  "playgroundai/playground-v2.5-1024px-aesthetic",
17
  torch_dtype=torch.float16,
 
10
 
11
  device = torch.device("cuda:0")
12
 
13
+ llm = AutoModelForCausalLM.from_pretrained("Azure99/blossom-v5-4b", torch_dtype=torch.float16, device_map="auto")
14
+ tokenizer = AutoTokenizer.from_pretrained("Azure99/blossom-v5-4b")
15
  diffusion_pipe = DiffusionPipeline.from_pretrained(
16
  "playgroundai/playground-v2.5-1024px-aesthetic",
17
  torch_dtype=torch.float16,