cbensimon HF staff commited on
Commit
ebfb93c
1 Parent(s): db4a1fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -41,14 +41,14 @@ if torch.cuda.is_available():
41
  "fluently/Fluently-XL-Final",
42
  torch_dtype=torch.float16,
43
  use_safetensors=True,
44
- )
45
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
46
 
47
 
48
  pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
49
- pipe.set_adapters("dalle")
50
 
51
- pipe.to("cuda")
52
 
53
 
54
 
 
41
  "fluently/Fluently-XL-Final",
42
  torch_dtype=torch.float16,
43
  use_safetensors=True,
44
+ ).to('cuda')
45
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
46
 
47
 
48
  pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
49
+ # pipe.set_adapters("dalle")
50
 
51
+ # pipe.to("cuda")
52
 
53
 
54