azhan77168 commited on
Commit
ad95a95
·
verified ·
1 Parent(s): 717af79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -6,9 +6,9 @@ import torch
6
  from PIL import Image
7
  from diffusers import StableDiffusionPipeline
8
  # Load the model
9
- # model_id = "nitrosocke/Ghibli-Diffusion"
10
  # model_id = "openfree/flux-chatgpt-ghibli-lora"
11
- # pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
12
 
13
 
14
 
@@ -25,8 +25,10 @@ from diffusers import StableDiffusionPipeline
25
  # lora_model_id = "openfree/flux-chatgpt-ghibli-lora"
26
  # pipe.load_lora_weights(lora_model_id)
27
 
28
- pipe = AutoPipelineForImage2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16,token=True)
29
- pipe.load_lora_weights('openfree/flux-chatgpt-ghibli-lora', weight_name='flux-chatgpt-ghibli-lora.safetensors')
 
 
30
 
31
  # Move pipeline to GPU if available
32
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
6
  from PIL import Image
7
  from diffusers import StableDiffusionPipeline
8
  # Load the model
9
+ model_id = "nitrosocke/Ghibli-Diffusion"
10
  # model_id = "openfree/flux-chatgpt-ghibli-lora"
11
+ pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
12
 
13
 
14
 
 
25
  # lora_model_id = "openfree/flux-chatgpt-ghibli-lora"
26
  # pipe.load_lora_weights(lora_model_id)
27
 
28
+ # pipe = AutoPipelineForImage2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16,token=True)
29
+ # pipe.load_lora_weights('openfree/flux-chatgpt-ghibli-lora', weight_name='flux-chatgpt-ghibli-lora.safetensors')
30
+ pipeline.load_lora_weights("alvarobartt/ghibli-characters-flux-lora")
31
+
32
 
33
  # Move pipeline to GPU if available
34
  device = "cuda" if torch.cuda.is_available() else "cpu"